eric612 / Mobilenet-YOLO-Pytorch

Include mobilenet series (v1,v2,v3...) and yolo series (yolov3,yolov4,...)

Mobilenet-YOLO-Pytorch

Model

A pytorch implementation of MobileNet-YOLO detection network , train on 07+12 , test on VOC2007 (imagenet pretrained , not coco)

Network	mAP	Resolution	download
MobileNetV2	72.1	352	checkpoint

Training steps

1. Download dataset VOCdevkit/ , if already have , please skip this step

sh scripts/VOC2007.sh
sh scripts/VOC2012.sh

2. Create lmdb

 sh scripts/create.sh 

3. Start training

sh scripts/train.sh 

yolov3 training

see branch

Hyper parameter optimization

nnictl create --config config.yml

Demo

Download checkpoint, and save at $Mobilenet-YOLO-Pytorch/checkpoints/bdd100k/model_best.pth.tar

sh scripts/inference.sh 

Under construction

• A new detector
• yolov4
• Multi-Task
• Hyper Parameter Tuning
• Pruning
• Porting KL720

Acknowledgements

AlexeyAB

diggerdu

BobLiu20

bubbliiiing

aleju

rmccorm4

hysts

utkuozbulak