Skip to content

Latest commit

 

History

History
 
 

object-detection

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!