Skip to content

developer0hye/YOLOv3Tiny

Repository files navigation

YOLOv3Tiny

Re-implementation PyTorch based YOLOv3Tiny

I tried to reproduce Darknet based YOLOv3Tiny using PyTorch.

Performance

I used the ImageNet pretrained backbone network.

Refer to this project to know training rules for training the pretrained network.

Training Set Test Set mAP(mean Average Precision) Weights
VOC 07+12 train/val VOC 07 test 58.86 download

Prepare Dataset

Use this script to prepare VOC dataset.

If you finished to download and convert labeling format, put the dataset folder into project folder.

├─backbone_weights
├─figures
├─test_example
├─VOCdataset
│  ├─test
│  └─train
| ...
| train.py
| test.py

Training Command

python train.py

Evaluation Command

python test.py --weights your_weight.pth