faster-RCNN配置(python接口)

前几天一口气看完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 --recursive
cd ~
git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git

3.编译Cython模块

cd $FRCN_ROOT/lib
make
这里$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
还不行就先con da 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
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值