Skip to content

Latest commit

 

History

History
20 lines (13 loc) · 702 Bytes

README.md

File metadata and controls

20 lines (13 loc) · 702 Bytes

YOLO Object Detection

This is a light-weight implementation of the YOLO v3 object detection model in PyTorch. It's been adapted specifically for the use case of counting cars in surface lots, but is general enough to work with object detection for all 80 classes in the COCO dataset.



YOLO PyTorch tutorial

Requirements

  • torch
  • numpy
  • opencv-python
  • Pre-trained weights file for YOLO v3: wget https://pjreddie.com/media/files/yolov3.weights

Run

  • python demo.py