
Pytorch中用于对象检测的特征金字塔网络(FPN)的实现。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
fpn.pytorch提供对象检测任务中的功能金字塔网络(FPN)的PyTorch实现,作为对我们r-cnn更快速PyTorch实现的一个延续。凭借此基础,该项目还具备以下显著特性:首先,它完全采用PyTorch代码编写,确保了高效性和可维护性。其次,我们已将所有原有的NumPy实现成功迁移至PyTorch框架中。此外,该项目支持训练批大小大于1的情况,这对于大规模数据集的训练尤为重要。为了进一步优化性能,我们对所有层进行了全面的修改和调整,包括数据加载器、RPN(区域提议网络)以及ROI pooling等关键模块,从而能够在每次迭代过程中利用多个图像进行训练。最后,该项目也支持多种GPU加速计算。
全部评论 (0)
还没有任何评论哟~


