← all repositories

amdegroot/ssd.pytorch

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

5.2k stars Python Computer VisionML Frameworks
ssd.pytorch
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.