faster-rcnn(python版本)安装
http://blog.csdn.net/u011635764/article/details/52831167
http://blog.csdn.net/u012841667/article/details/53436615
安装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
然后就大功告成了.
如遇到 这个问题 make: *** [.build_release/src/caffe/layers/hdf5_output_layer.o] Error 1
修改Makefile.config需要修改的内容：
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial