PyTorch-YOLOv4

This project aims to implement YOLOv4 in PyTorch.

Paper: https://arxiv.org/pdf/2004.10934.pdf

Original Implementation and Source Code: https://github.com/AlexeyAB/darknet

This is still a work in progress.

Getting Started

Prerequisites

This project of yolov4 supports Python >= 3.6 and Torch >= 1.5.0

Installing

Clone this repo by running

git@github.com:jason-hu7/PyTorch-YOLOv4.git

cd to the directory and install from source

cd PyTorch-YOLOv4
pip install .

Training

How to train a model

python train.py

Testing

How to test and evalute a model

Place holder

Detection

Use trained model to detect objects in images

Place holder

Contributing

No guidelines yet, help a brother out

License

This project is licensed under the MIT License - see the LICENSE.md file for details