← all repositories

lufficc/SSD

A modular PyTorch implementation of SSD (Single Shot MultiBox Detector) for object detection with support for multiple backbones and multi-GPU training.

1.6k stars Python Computer VisionML Frameworks
SSD
Velocity · 7d
+0.6
★ / day
Trend
steady
star history

This repository provides a high-quality, fast reference implementation of the SSD object detection algorithm in PyTorch. The implementation is modular, allowing researchers to easily swap components like backbones (VGG, EfficientNet), detectors, and box predictors. It supports distributed training across multiple GPUs, batched inference, evaluation during training, and TensorBoard visualization of metrics.

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