This repo implements SSD (Single Shot MultiBox Detector) in PyTorch for object detection, using MobileNet backbones. It also has out-of-box support for retraining on Google Open Images dataset and Pascal VOC.
For documentation, please refer to the following:
- Re-training SSD-Mobilenet from the Hello AI World tutorial
- Original Readme from
https://github.com/qfgaohao/pytorch-ssd
Thanks to @qfgaohao for the upstream implementation