Skip to content

Files

Latest commit

e2ca474 · Jan 29, 2020

History

History
This branch is 445 commits behind x4nth055/pythoncode-tutorials:master.

object-detection

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
Jan 29, 2020
Jan 29, 2020
Jan 29, 2020
Jan 29, 2020
Jan 29, 2020
Jan 29, 2020
Jan 29, 2020
Jan 29, 2020
Jan 29, 2020
Jan 29, 2020
Jan 29, 2020

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!