#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系列是怎么运转的,而不是拿来直接部署的成品。
相关文章
暂无针对该项目的文章。
如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。