pytorch-yolov4's Introduction

Pytorch-YOLOv4

A minimal PyTorch implementation of YOLOv4.

Paper Yolo v4: https://arxiv.org/abs/2004.10934
Source code:https://github.com/AlexeyAB/darknet
More details: http://pjreddie.com/darknet/yolo/
Inference
Train
- Mocaic

├── README.md
├── dataset.py      dataset
├── demo.py         demo --> tool/darknet2pytorch
├── models.py       model for pytorch
├── train.py        train models.py
├── cfg.py          cfg.py for train
├── cfg             cfg --> darknet2pytorch
├── data            
├── weight          --> darknet2pytorch
├── tool
│   ├── camera.py           a demo camera
│   ├── coco_annotatin.py       coco dataset generator
│   ├── config.py
│   ├── darknet2pytorch.py
│   ├── region_loss.py
│   ├── utils.py
│   └── yolo_layer.py

Weight

baidu(https://pan.baidu.com/s/1dAGEW8cm-dqK14TbhhVetA Extraction code:dm5b)
google(https://drive.google.com/open?id=1cewMfusmPjYWbrnuJRuKhPMwRe_b9PaT)

Inference

download model weight https://drive.google.com/open?id=1cewMfusmPjYWbrnuJRuKhPMwRe_b9PaT

python demo.py cfgfile weightfile imgfile

Reference:

@article{yolov4,
  title={YOLOv4: YOLOv4: Optimal Speed and Accuracy of Object Detection},
  author={Alexey Bochkovskiy, Chien-Yao Wang, Hong-Yuan Mark Liao},
  journal = {arXiv},
  year={2020}
}

Recommend Projects