this repo is forked from https://github.com/amdegroot/ssd.pytorch. Implemented by pytorch.
Contributions:
- implement mobielentv2 for ssd.
- add focal loss. (need adjuct super-parameters).
- add detection.py demo for image and video detection.
result(train on voc 2007trainval + 2012, test on voc 2007test):
- ssd-mobielnetv2 (this repo): 70.27%. (without focal loss).
- ssd-mobielentv1: 68.% (without COCO pretaining), 72.7% (with COCO pretraining) https://github.com/chuanqi305/MobileNet-SSD.
- ssd-vgg16 (paper): 77.20%.
pretrained model and trained model: