110awesome / yolov3_pytorch

YOLOV3 implementation with pytorch using pytorch approch to define the network

YOLO v3 implementation With pytorch

this repository only contain the detection module and we don't need the cfg from original YOLOv3, we implement it with pytorch.

This repository is based on the official code of YOLOv3 and pytorch-yolo-v3. There's also an implementation for YOLOv3 already for pytorch, but it uses a config file rather than a normal pytorch approch to defining the network. One of the goals of this repository is to remove the cfg file.

Requirements

• Python 3.5
• OpenCV
• PyTorch 0.4

Installation

• Install PyTorch-0.4.0 by selecting your environment on the website and running the appropriate command.
• Clone this repository
• Compile the nms
• convert yolov3.weights to pytorch

cd YOLOv3_Pytorch
./make.sh

mkdir weights
cd weights
wget https://pjreddie.com/media/files/yolov3.weights
python convert_darknet.py 
# we will get the convert_yolov3.pth

Demo

python demo.py --images images --save_path ./output --weights ./weights/convert_yolov3.pth

Example

References

• YOLOv3: An Incremental Improvemet

• Original Implementation (Darknet)

• pytorch-yolo-v3

• pytorch-yolo2