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) |