#326 · 主分类: 计算机视觉

ssd.pytorch

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

PyTorch实现的单次多框检测器

项目最后更新:12/29/21

GitHub Stars

5.2K

Forks数量

1.7K

贡献者数量

11

许可证

MIT

收录理由

这是 SSD(Single Shot MultiBox Detector)在 PyTorch 中的参考实现,源自 2016 年的论文。它通过单次前向传播同时预测边界框和类别分数,让实时检测成为可能。如果你想了解现代检测器的实际构造,这份代码值得一读:VGG 骨干网络、额外的 multibox 层以及损失处理各自独立成模块,与论文对应紧密。项目还提供了 Pascal VOC 和 COCO 的数据集加载器及下载脚本,并集成了 Visdom,可在浏览器中实时观察训练损失。需要注意的是,该项目自 2021 年底起不再维护,面向较旧版本的 PyTorch API,因此更适合作为清晰的教学参考和复现论文结果的起点,而不是直接用于新生产系统的依赖。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目