SSD-based Object Detection in PyTorch

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.

For documentation, please refer to Object Detection portion of the Hello AI World tutorial: Re-training SSD-Mobilenet

Thanks to @qfgaohao for the upstream implementation from: https://github.com/qfgaohao/pytorch-ssd