PyTorch implementation of YOLOv3 object detector

This is an implementation of YOLOv3 object detector using PyTorch. This code dependes on the darknet implementation of YOLOv3. The official darknet website can be found here.

Installation

Install requirements

$ git clone https://github.com/MoaazAbdulrahman/PyTorch-YOLOv3-Object-detector
$ cd PyTorch-YOLOv3-Object-detector/
$ sudo pip install -r requirements.txt

Download model weights here and add it to weights folder

Detection Example

Video processing

processing.py file is to be used for video processing. It contains processing() method which is responsible for inference. I have used YOLOv3 for person detection and counting. Follow this link to watch the output video.