#326 · Primary category: Computer Vision

ssd.pytorch

computer-vision deep-learning image-recognition machine-learning object-detection pytorch ssd webcam

A PyTorch Implementation of Single Shot MultiBox Detector

Project last updated:12/29/21

GitHub Stars

5.2K

Forks

1.7K

Contributors

11

License

MIT

Why we included this project

This is the reference PyTorch implementation of SSD, the 2016 Single Shot MultiBox Detector that made real-time detection practical by predicting bounding boxes and class scores in a single forward pass. If you want to learn how modern detectors are actually built, the code reads well: the VGG backbone, the extra multibox layers, and the loss handling each sit in their own modules and follow the paper closely. It also includes dataset loaders for Pascal VOC and COCO with download scripts, plus a Visdom integration that lets you watch training loss live in the browser. Be aware the project has been unmaintained since late 2021 and targets an older PyTorch API, so it works best as a clear educational reference and a starting point for reproducing paper results, not as a drop-in dependency for new production systems.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category