faster-RCNN配置(python接口)

来源：互联网发布：天一混元青岛店淘宝编辑：程序博客网时间：2024/06/05 01:52

前几天一口气看完rbgirshick大神的三篇物体检测的论文还有spp-net顺便也看完了，不得不佩服大神对前沿技术的明锐眼神，于是终于在今天成功运行好faster-RCNN的demo。

1.配置caffe，建议先按照我之前的caffe配置教程配置下原版的caffe，不然很多概念会不懂，而且先配置好原版caffe，后面faster-caffe的配置要省去不少步骤。

caffe安装流程和遇到问题

2.下载faster-RCNN

去RBG大神的guthub下载faster-cnn

# Make sure to clone with --recursivecd ~git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git

3.编译Cython模块

cd $FRCN_ROOT/libmake

这里$FRCN_ROOT表示faster-RCNN的根目录。原版caffe没有这一步，其实这里主要是把lib文件夹下的py文件导入python环境变量中，以便在命令行中使用faster-RCNN

4.编译Caffe和pycaffe

cd $FRCN_ROOT/caffe-fast-rcnn# Now follow the Caffe installation instructions here:#   http://caffe.berkeleyvision.org/installation.html# If you're experienced with Caffe and have all of the requirements installed# and your Makefile.config in place, then simply do:make -j8 && make pycaffe

这里和原版caffe没有区别，包括Makefile.config文件，可以直接拿过来用。如果库都安装好没问题，报未定义的引用错误时建议加上sudo试试。

5.下载faster-RCNN的网络参数文件

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

点开脚本文件，可以看到是从dropbox上下载，dropbox被墙了，建议自己去百度云下载，记得别下错了下了fast-RCNN，下完后解压放在./data文件夹下，即VGG16_faster_rcnn_final.caffemodel的目录是./data/faster_rcnn_models

faster_rcnn_models，密码：gbpo

6.依赖库

这时候离运行demo就还差一步了，也是最重要的一步，你的python需要cython，python-opencv和easydict包。

6.1.cython

如果你下的是anaconda那么cython就是自带的。

6.2.easydict

执行

pip install easydict

如果没装pip就换成

conda install easydict

还不行就先conda install pip安装pip后在用pip安装easydict，这里我遇到一个报错，无法找到easydict，建议更新下pip就好pip install pip --upgrade

6.3.python-opencv

头疼的是python-opencv包，demo中有用到import cv2，如果在运行demo的时候报错，No module named cv2就是这个问题，我们需要的就是这个cv2.so文件，这个文件需要安装并编译opencv后才会在/usr/local/lib/python2.7/site-packages出现。

opencv下载官网

我下的是3.2.0

编译好后将/usr/local/lib/python2.7/site-packages下的cv2.so复制到~/anaconda2/lib/python2.7/site-packages，如果你没有用anaconda那就复制到/usr/lib/python2.7/site-packages/

7.demo

记得在~/.basrc中把python的环境变量改了，不然import caffe导入的是原版的caffe，在export PYTHONPATH="/home/yourname/caffe-master/python:$PYTHONPATH"添加#注释掉，增加export PYTHONPATH="/home/yourname/py-faster-cnn/caffe-fast-rcnn/python:$PYTHONPATH"，这样也方便，一台电脑两个caffe框架，想用哪个改哪个

cd $FRCN_ROOT./tools/demo.py

0 0