Search code examples
machine-learningcomputer-visionobject-detectionyolodarknet

YOLO (Darknet): How to detect a whole directory of images?


The Darknet guide to detect objects in images using pre-trained weights is here: https://pjreddie.com/darknet/yolo/

The command to run is:

./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg

The last argument is the path to a file, I've tried to change it to data/*.jpg but didn't work.

How to use Darknet to detect a whole directory of images?


Solution

  • As per the link mentioned below, one can use cv2.dnn.readNetFromDarknet module to read darknet, trained weights and configuration file to make a loaded model in python. Once the model is loaded, one can simply use for loop for prediction. Please refer this link for further clarification