
Faster-R-CNN是一种快速的目标检测算法。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
令人振奋的消息!现在,这个仓库已经成功地支持了PyTorch 1.0版本! 我们借鉴并吸收了部分代码和技术。 该项目专注于提供更快速的R-CNN的PyTorch实现,其核心目标是显著提升R-CNN对象检测模型的训练速度。 近期涌现出诸多优秀的实现方案,例如基于Pycaffe + Numpy的实现、基于Pytorch + Numpy的实现、以及基于TensorFlow + Numpy的实现。 在我们的开发过程中,我们积极参考并借鉴了这些现有实现,尤其是在...方面。 然而,相较于这些已有的实现,我们的版本拥有诸多独特且创新性的功能:首先,它完全采用纯PyTorch代码构建;其次,我们已将所有基于Numpy的实现全部转化为PyTorch版本。此外,该项目还具备多图像批处理训练的支持能力,并通过对数据加载器、RPN(Region Proposal Network)以及ROI池等所有层进行修改,从而能够在每个MiniBatch中同时处理多个图像。 最后,该项目还支持利用多个GPU进行训练。
全部评论 (0)
还没有任何评论哟~


