Pytorch版faster rcnn的安装、配置与测试

Pytorch作为一个较新的开源框架，十分简洁好用，完全不亚于Tensorflow等成熟框架。

最近在学习Faster R-CNN, 发现Pytorch版本的资料不多，所以在这里记录与分享下自己安装配置Pytorch版本的faster cnn的过程。过程是搬运+翻译的说明文档，增加了一些细节性的东西和报错解决。

这里我用的是github上ruotianluo开源的代码:pytorch-faster-rcnn

一、安装

• 下载代码

git clone https://github.com/ruotianluo/pytorch-faster-rcnin.git

• 编译模块(nms, roi_pooling, roi_align)

这里可能会报错，后面讲解决方案

cd pytorch-faster-rcnn/lib

bash make.sh

cd ../

• 安装python COCO API （获取COCO数据集用）

cd data

git clone https://github.com/pdollar/coco.git cd coco/PythonAPI

make

cd ../../..

编译的过程中可能会报错，这时候对照下这个表，去 make.sh 里删掉的不支持的arch

GPU model	Architecture
TitanX (Maxwell/Pascal)	sm_52
GTX 960M	sm_50
GTX 1080 (Ti)	sm_61
Grid K520 (AWS g2.2xlarge)