更快的RCNN-pytorch
FasterRCNN在VGG,ResNet和FPN基础中实现。
参考:
rbg的FasterRCNN代码: :
模型表现
在VOC2017上进行培训在VOC2017上进行测试
骨干
地图
VGG16
0.7061
ResNet101
0.754
训练模型
1.运行前,您需要:
光盘./lib
在make.sh和setup.py中更改gpu_id。 具体来说,您需要在make.sh的第5、12和19行以及setup.py的第143行中修改参数设置,其中包含关键字“ -arch =”取决于您的gpu模型。(选择适当的体系结构,见下表)
sh make.sh
GPU型号
建筑学
TitanX(麦克斯韦/帕斯卡)
sm_52
GTX 960M
sm_50
GTX 108(钛)
sm_61
网格K520(AWS g2.2xlarge
1