Skip to content

Files

Latest commit

Oct 11, 2024
8c96f0f · Oct 11, 2024

History

History

object-detection

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
Oct 11, 2024
Oct 11, 2024
Oct 1, 2023
Oct 11, 2024
Oct 11, 2024
Oct 11, 2024
Oct 11, 2024
Oct 11, 2024
Oct 11, 2024
Oct 11, 2024
Oct 11, 2024

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!