CaesarObjectDetection/README.md

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!