offlinemot.detection.YoloDetector.detect

YoloDetector.detect(imgfile)[source]

Detect the classes within the image

Parameters

imgfile (str or numpy array) – The input image that could be a string filename or a numpy array

Returns

A list of detection positions and classes and the shape of the original image. The list has the following shape [top-left point, bottom-right point,probabilty, class id]

Return type

list,(int,int)