dusty-nv / pytorch-ssd

MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in PyTorch. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.

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