pytorch-ssd
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:
-
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