amdegroot/ssd.pytorch
A PyTorch implementation of the Single Shot MultiBox Detector architecture for real-time object detection.

Velocity · 7d
+1.5
★ / day
Trend
→steady
star history
This repository provides a PyTorch reimplementation of the SSD object detection model from the 2016 paper by Liu et al. It supports training and evaluation on VOC and COCO datasets, includes a dataset loader compatible with torch.utils.data.Dataset, and offers real-time loss visualization via Visdom during training. The implementation reproduces the original Caffe-based architecture in modern PyTorch.