caffe学习一：ubuntu16.04下跑Faster R-CNN demo (基于caffe). (亲测有效，记录经历两天的吐血经历)

兜兜转转，兜兜转转;

一次有一次，这次终于把Faster R-CNN 跑通了。

 

重要提示1：在开始跑Faster R-CNN之前一定要搞清楚用的是Python2 还是Python3. 不然你会无限次陷入一下错误：

from ._caffe import Net, SGDSolver, NesterovSolver, AdaGradSolver,
ImportError: dynamic module does not define module export function (PyInit__caffe) 

错误原因：当出现这个错误的时候，你应该知道错误原因极有可能是Python版本问题。

首先，ubuntu是自带两个版本的Python, 通常做深度学习的人会装anaconda.这样算下来你的计算机就有三个版本的python了：

python2.7, python3.5, conda版本的python (具体版本可以终端查看。)

 

好了，现在你知道你电脑有若干个版本的python. 那么跑Faster R-CNN应该用哪个版本呢？

（个人建议用官方自带的python2.7。 在github上下载的也是2.x版本的程序。以下教程也是2.7版本的。）

 

重要提示2：如果你用Python2.x跑Faster R-CNN,并且你装了anaconda。那么你首先要做的就是把anaconda的环境变量路径给注释掉。

步骤如下：(1)终端输入：gedit ~/.bashrc

                 (2)然后将anaconda的路径注释掉(如下图，在export前面加一个#.然后右上角保存并关闭退出。)：

                   

 

 

                (3)最后输入source ~/.bashrc使bashrc文件的修改生效

 

                  

 

 

 

 

重要提示3：查看下你电脑默认的是python2还是python3.

 

在终端输入python:

 

 

 

 如图所示，我的已经切换至python2.x。 如果你的是Python3.x，直接执行以下命令即可：

sudo update-alternatives --install /usr/bin/python python /usr/bin/python2 100

sudo update-alternatives --install /usr/bin/python python /usr/bin/python3 150

(借图)

直接命令：python，默认为python3,如下

如果要切换到Python2，执行：sudo update-alternatives --config python

 

 

然后输入对应的数字即可。

再一次输入python

 

这就成功i切换到Python2.x啦。当然了，你一开始的就是Python2.x。 就不用走以上步骤。

 

下面开始愉快的跑Faster R-CNN. 皮皮虾，跟我走！！

 

一，下载文件

下载faster R-CNN，在终端下执行命令：git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git

 

二，编译caffe

 

1、进入py-faster-rcnn文件夹下的lib文件夹内

(1). cd py-faster-rcnn/lib

(2). make

 

2、进入py-faster-rcnn文件夹下的caffe-fast-rcnn下，把Makefile.config.example拷贝一份重命名为Makefile.config文件，命令如下：
(1). cp Makefile.config.example Makefile.config
(2). 修改Makefile.config下的相关参数: sudo gedit Makefile.config

根据个人情况修改文件：
a.若使用cudnn，则将
#USE_CUDNN := 1
修改成：
USE_CUDNN := 1

b.若使用的opencv版本是3的，则 将
#OPENCV_VERSION := 3
修改为：
OPENCV_VERSION := 3

c.若要使用python来编写layer，则 将
#WITH_PYTHON_LAYER := 1
修改为
WITH_PYTHON_LAYER := 1

d.重要的一项 : 将 # Whatever else you find you need goes here. 下面的

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib
修改为：

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
这是因为Ubuntu16.04的文件包含位置发生了变化，尤其是需要用到的hdf5的位置，所以需要更改这一路径.

(4)修改makefile文件 打开makefile文件，做如下修改：
将：
NVCCFLAGS +=-ccbin=$(CXX)-Xcompiler-fPIC $(COMMON_FLAGS)
替换为：
NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)

(5)编辑/usr/local/cuda/include/host_config.h

将其中的第115行注释掉： 将

# error-- unsupported GNU version! gcc versions later than 4.9 are not supported!
改为
//#error-- unsupported GNU version! gcc versions later than 4.9 are not supported!

 

3. 配置完成后，下面进行编译。直接编译会出现编译错误，这是因为faster-rcnn默认的caffe支持的cudnn版本是V4，而想用1080Ti显卡，必须使用cuda7.5+cudnn5.0以上版本，因此编译caffe会出现版本不兼容而导致的函数参数不对应的错误。

解决办法，下载最新的caffe：https://github.com/BVLC/caffe

(1)、将caffe/include/caffe/util/cudnn.hpp拷贝到py-fast-rcnn/caffe-fast-rcnn/include/caffe/util/

(2)、将caffe/include/caffe/layers里所有以cudnn开头的文件都拷贝至py-fast-rcnn/caffe-fast-rcnn/include/caffe/layers,全部替换掉

(3).将caffe/src/caffe/layers里所有以cudnn开头的文件都拷贝至py-fast-rcnn/caffe-fast-rcnn/src/caffe/layers,全部替换掉

改好后先 make clean

再重新执行 make

不再报错

再执行

make pycaffe

 

三. 下载预训练模型：

1  ./data/scripts/fetch_faster_rcnn_models.sh

2  ./data/scripts/fetch_imagenet_models.sh

下载完成后data目录会有对应的模型文件存在

 

四. 运行demo.py

1. cd py-faster-rcnn/tools

2. ./demo.py                            # 该命令默认的是VGG16的模型

或者执行

./demo.py --net zf     #执行此命令可以运行zf模型

 

效果如下图：

 

 

贴一个很有用的帖子：https://blog.csdn.net/flygeda/article/details/78638824