文件名称:Faster-R-CNN
文件大小:242.68MB
文件格式:ZIP
更新时间:2024-05-12 04:54:17
Python
更快的R-CNN的更快的Pytorch实现 介绍 好消息! 这个仓库现在支持pytorch-1.0 !!! 我们从借了一些代码和技术。 此项目是更快的R-CNN的更快的pytorch实现,旨在加速训练更快的R-CNN对象检测模型。 最近,有许多不错的实现: ,基于Pycaffe + Numpy开发 ,基于Pytorch + Numpy开发 基于TensorFlow + Numpy开发的 ,基于Pytorch + TensorFlow + Numpy开发 在实现过程中,我们引用了上述实现,尤其是 。 但是,与上述实现相比,我们的实现具有一些独特的新功能: 它是纯Pytorch代码。 我们将所有的numpy实现都转换为pytorch! 它支持多图像批处理训练。 我们修改了所有层,包括数据加载器,rpn,roi池等,以在每个minibatch中支持多个图像。 它支持多个GPU训