#326 · 主分类: 计算机视觉
ssd.pytorch
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,因此更适合作为清晰的教学参考和复现论文结果的起点,而不是直接用于新生产系统的依赖。
相关文章
暂无针对该项目的文章。
如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。