Faster RCNN python 安裝

Faster RCNN python 安裝

轉載 2016年11月23日 11:11:57

• 3048

原文轉自http://blog.csdn.net/u011635764/article/details/52831167

這篇博客主要是記錄如何安裝 python版的faster-RCNN 
,因爲之前已經安裝過caffe了, 
Ubuntu16.04系統,顯卡GTX1060以各種依賴,cuda8.0, cudnn, MKL, opencv3.1.0, 
都已經安裝過了,python環境是Anaconda,之後有機會補上前面的各種坑吧. 
另外歡迎訪問我的博客,實在是百度不到的說 …. http://hammer2505.cn/ 

安裝python依賴包

sudo pip install Cython
sudo pip install easydict

• 1
• 2

下載源代碼

git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git

• 1

修改並編譯Cython

cd py-faster-rcnn/lib

• 1

修改setup.py文件第135行

'nvcc': ['-arch=sm_35',

• 1

根據自己的顯卡計算能力進行修改,查詢網址爲:CUDA GPUs

例如我的電腦是GTX 1060, 計算能力爲6.1,則修改爲

'nvcc': ['-arch=sm_61',

• 1

保存後執行

make

• 1

進行編譯.

編譯caffe

修改Makefile.config文件

cd ../caffe-faster-rcnn

• 1

進入caffe目錄. 
我是將自己之前的caffe的Makefile.config拷貝到了該目錄下,也可以執行

cp Makefile.config.example Makefile.config

• 1

對Makefile.config文件進行修改. 
注意修改

USE_CUDNN := 1
WITH_PYTHON_LAYER := 1
USE_PKG_CONFIG := 1

• 1
• 2
• 3

修改cudnn加速文件

這裏需要注意一下,因爲作者的caffe版本不支持新的cudnn加速,我們需要將

py-faster-rcnn/caffe-fast-rcnn/include/caffe/util/cudnn.hpp  
py-faster-rcnn/caffe-fast-rcnn/src/caffe/util/cudnn.cpp
py-faster-rcnn/caffe-fast-rcnn/src/caffe/layers/ 目錄下16個cudnn_開頭的文件
py-faster-rcnn/caffe-fast-rcnn/include/caffe/layers/目錄下8個cudnn_開頭的文件

• 1
• 2
• 3
• 4

選擇caffe-master版的文件進行替換,這樣編譯就不會出錯啦.

修改一個include錯誤

修改

py-faster-rcnn/caffe-fast-rcnn/src/caffe/test/test_smooth_L1_loss_layer.cpp

• 1

刪除第十一行

#include "caffe/vision_layers.hpp"

• 1

編譯caffe

# compile Caffe modules
make -j8 all  

make -j8 test  

# run tests to ensure all the tests are passed
make runtest  

# build pycaffe  
make pycaffe

• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9
• 10

下載已經訓練好的模型

cd $FRCN_ROOT
./data/scripts/fetch_faster_rcnn_models.sh

• 1
• 2

執行Demo

cd $FRCN_ROOT
./tools/demo.py

• 1
• 2

然後就大功告成了.