软件环境:
Ubuntu 16.04 + CUDA8.0 + cuDnn5.1 + python 2.7 + OpenCv 3.1
本文的主要目的,是解决在编译py-faster-rcnn的过程中,与cuDnn的v5版本的冲突问题。编译报错是函数错用。最初是把cuDnn换成了v4。后期在跑demo.py时,没能正确检测出物体,也就是没有出带框的图像。一开始以为是plt的问题,后来发现不是,此demo.py在cpu下运行正常,加上gpu选项就不能正常检测物体。
第一阶段,是按照py-faster-rcnn作者的前半部分步骤来进行。相关链接
1.作者提到,在编译Caffe时,至少需要在Makefile.config设置的两点,这里和Caffe的安装相关了,有很多这样的教程。
# In your Makefile.config, make sure to have this line uncommented
WITH_PYTHON_LAYER := 1
# Unrelatedly, it's also recommended that you use CUDNN
USE_CUDNN := 1
2.你需要安装的软件(用apt-get即可):
cython, python-opencv, easydict
3.命令行下
# Make sure to clone with --recursive
git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git
4. $FRCN_ROOT指克隆过来的根目录
cd $FRCN_ROOT/lib
make
==========================================================================分界线=======================