A PyTorch implementation of a YOLO v3 Object Detector

This repository contains code for a object detector based on YOLOv3: An Incremental Improvement, implementedin PyTorch. The code is based on the official code of YOLO v3, as well as a PyTorch port of the original code, by marvis. One of the goals of this code is to improve upon the original port by removing redundant parts of the code (The official code is basically a fully blown deep learning library, and includes stuff like sequence models, which are not used in YOLO). Modified the original code forked from Modified the original code forked ayooshkathuria/pytorch-yolo-v3

Sample Usage

python sample_detctor.py --input <input_video> --output <output_path>

Setup

git clone https://github.com/sumitbinnani/pytorch-implementation-yolo-v3.git


# Download weights
mkdir weights
cd weights
wget https://pjreddie.com/media/files/yolov3.weights