← all repositories

yatengLG/SSD-Pytorch

A PyTorch implementation of the SSD object detection algorithm with multi-GPU training support.

402 stars Python Computer VisionML Frameworks
SSD-Pytorch
Velocity · 7d
+0.2
★ / day
Trend
steady
star history

This repository provides a complete implementation of the Single Shot MultiBox Detector (SSD) algorithm for object detection using PyTorch. It features an object-oriented architecture separating base networks, detection heads, loss functions, and post-processing modules. The project supports multi-GPU training on a single machine and includes methods for detecting objects in both images and videos. All code is documented with Chinese comments for easier understanding.

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