CaesarCloudSync
CaesarAI Deployed
9d3162f
|
raw
history blame
1.05 kB

How to Perform YOLO Object Detection using OpenCV and PyTorch in Python

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:
    python yolo_opencv.py images/dog.jpg
    
    A new image 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:
    python read_video.py video.avi
    
    This will start detecting objects in that video, in the end, it'll save the resulting video to 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!