Mobilenet-YOLO-Pytorch

Model

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

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
• [x] yolov4
• [x] Multi-Task
• [x] Hyper Parameter Tuning
• [ ] Pruning
• [x] Porting KL720

Acknowledgements

AlexeyAB

diggerdu

BobLiu20

bubbliiiing

aleju

rmccorm4

hysts

utkuozbulak