python版本的faster-rcnn见我的另一篇博客:
py-faster-rcnn(running the demo): ubuntu14.04+caffe+cuda7.5+cudnn5.1.3+python2.7环境搭建记录
1. 首先需要配置编译caffe的环境,并降级gcc为4.7.见: ubuntu14.04下安装cudnn5.1.3,opencv3.0,编译caffe及matlab和python接口过程记录(不好意思,这也是我自己写的)
2. clone 源码:
git clone --recursive https://github.com/ShaoqingRen/faster_rcnn
3. clone 作者的caffe源码(记住一定要是作者的,不然运行matlab程序时会出错,我想这个道理应该很明白).不过这一步应该在上面的recursive clone做到了.
4. 在作者提供的百度云链接上下载训练好的模型,当然也可以"Run fetch_data/fetch_faster_rcnn_final_model.m
to download our trained models",不过速度会很慢.
5. 进入faster_rcnn/external/caffe,复制一份之前编译caffe时的Makefile.config,也可以复制当前文件夹下的Makefile.config.example,去掉.example后缀.
cd external/caffe
6. 修改Makefile.config文件,加入matlab路径.我的.config文件重要部分如下:
#USE_CUDNN :=
OPENCV_VERSION :=
CUDA_DIR := /usr/local/cuda
BLAS := atlas
MATLAB_DIR := /usr/local/MATLAB/R2014a
ANACONDA_HOME := $(HOME)/anaconda2
PYTHON_INCLUDE := $(ANACONDA_HOME)/include \
$(ANACONDA_HOME)/include/python2. \
$(ANACONDA_HOME)/lib/python2./site-packages/numpy/core/include \
PYTHON_LIB := $(ANACONDA_HOME)/lib
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib USE_PKG_CONFIG :=
值得注意的是这里好像用不到cudnn,用了反而会报错,试了各种办法都不行。。。
7 开始编译caffe和matlab接口
make clean
make -j8
make matcaffe
8 按照作者提供的testing步骤跑demo:
- Run
faster_rcnn_build.m
- Run
startup.m
- Run
experiments/script_faster_rcnn_demo.m
to test a single demo image.
9 到此,算是功德圆满.大致结果如下: