Mobilenet-YOLO-Pytorch

Like my previous project Mobilenet-YOLO , the loss function was very simular to original implemention of darknet

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/checkpoint/

sh scripts/inference.sh 

Under construction

• A new detector
• yolov4
• Multi-Task
• Hyper Parameter Tuning
• Pruning
• Porting (KL520 , ncnn , caffe , ...)

Acknowledgements

AlexeyAB

diggerdu

BobLiu20

bubbliiiing

aleju

rmccorm4

hysts

utkuozbulak