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

faster_rcnn_pytorch

computer-vision detection faster-rcnn pytorch

基于PyTorch的Faster RCNN

项目最后更新:09/25/21

GitHub Stars

1.8K

Forks数量

460

贡献者数量

4

许可证

MIT

收录理由

这是一份用PyTorch从零手写的Faster R-CNN移植代码,作者当初边学PyTorch边写,并没有刻意追求生产级质量。项目没有直接调用高层封装库,而是把区域提议网络和RoI池化一层层亲手搭起来,包括锚点生成、提议打分,以及用CUDA和Cython写的NMS与池化层——这些细节对想弄清两阶段检测器内部原理的人很有价值,适合当学习材料来读。作者在README里也很坦白,说精度没达到原版Caffe的水平,且只适配旧版PyTorch,并推荐了后续维护更好的分支用于实际训练。所以它更像一份架构拆解参考,帮你理解R-CNN系列是怎么运转的,而不是拿来直接部署的成品。

相关文章

暂无针对该项目的文章。

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

同类热门项目