To run this:
pip3 install -r requirements.txt
- Download the model weights and put them in
weights
folder. - To generate a object detection image on
images/dog.jpg
:A new imagepython yolo_opencv.py images/dog.jpg
dog_yolo3.jpg
will appear which has the bounding boxes of different objects in the image. - For live object detection:
python live_yolo_opencv.py
- If you want to read from a video file and make predictions:
This will start detecting objects in that video, in the end, it'll save the resulting video to
python read_video.py video.avi
output.avi
- If you wish to use PyTorch for GPU acceleration, please install PyTorch CUDA here and use
yolo.py
file. - Feel free to edit the codes for your needs!