pytorch-ssd icon indicating copy to clipboard operation
pytorch-ssd copied to clipboard

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:

Thanks to @qfgaohao for the upstream implementation