FastRCNN ubuntu14.04 GPU下配置

Faster R-CNN-python ubuntu14.04 GPU下配置

最近想开始学习一下深度学习的开源库,便以fasterRCNN进行入手学习。废话不多说,开始安装。


1.安装caffe所需的依赖项

需要说明一下,caffe是一个开源库,而fasterRCNN-py则是基于caffe源码修改后的一个实现fasterRCNN的库,因此fasterRCNN本身运行与官方caffe库没关系,只要编译修改版的fasterRCNN-py即可。而fasterRCNN-py和caffe源码所依赖的环境大致是一样的,因此我是按照caffe安装好后,再编译fasterRCNN这个源码。这里不考虑matlab接口的版本。

1.1安装CUDA

参考linux公社
http://www.linuxidc.com/Linux/2015-07/120449.htm Caffe在Ubuntu 14.04 64bit 下的安装

1.2安装CUDNN

参考linux公社
http://www.linuxidc.com/Linux/2015-07/120449.htm Caffe在Ubuntu 14.04 64bit 下的安装

1.3 Caffe原版安装

见linux公社的第六步,就是对源码里的配置文件进行设置,根据自己的机器。最好有GPU,另外就是得开启支持python的选项:WITH_PYTHON_LAYER := 1
然后再安装一些依赖项:
sudo apt-get install python-numpy python-scipy python-matplotlib python-sklearn python-skimage python-h5py python-protobuf python-leveldb python-networkx python-nose python-pandas python-gflags cython ipython
sudo apt-get update
我这里用的是python2.7,不是ancona的python。配置好之后就进行编译:
make all -j4
make test
make runtest
make pycaffe

1.4 Caffe环境变量配置

这个步骤主要是把原版的caffe添加到系统的路径里,同时也可以检查python是否能够导入caffe:
sudo gedit ~/.bashrc
添加 export PYTHONPATH=/home/startag/caffe/python/ 注意不需要添加其他路径了,不然出错。
最后在主目录下进入python,调用import caffe,看有没有错误。

2. py-faster-rcnn安装

这里主要参考官方github上的安装步骤 https://github.com/rbgirshick/py-faster-rcnn

2.1克隆源码

git clone –recursive https://github.com/rbgirshick/py-faster-rcnn.git

2.2修改配置

https://github.com/rbgirshick/py-faster-rcnn 上面介绍了如何修改配置,这里如果按照它说的方式修改Makefile.config文件,或者下载它给的,我最后编译是能通过,但是运行出错了,python报出corrupted double-linked list这个错误,在github的issue里也有,不过没人回答。尝试了好久,最后把caffe源码编译时的makefile配置文件复制过来,才没错误的。

2.3安装依赖项

官网说要安装cython, python-opencv, easydict这三个。caffe安装的依赖项有前两个,最后一个安装一下就行。

2.4编译Cython

cd $FRCN_ROOT/lib
make

2.5文件替换

主要是将fasterRCNN-py源码里caffe-faster-rcnn/include/caffe/layers/和caffe-faster-rcnn/src/caffe/layers路径下,cudnn开头的文件,用官方caffe源码里的对应文件替换掉。由于这个源码里的caffe部分在修改时较老,需要更新一下,不然会出错。

2.6编译源码

cd $FRCN_ROOT/caffe-fast-rcnn
make -j8 && make pycaffe

2.7下载Faster R-CNN detectors

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

2.8运行Demo

cd $FRCN_ROOT
./tools/demo.py

运行结果

这里写图片描述

参考

caffe安装

http://www.linuxidc.com/Linux/2015-07/120449.htm
http://www.cnblogs.com/empty16/p/4828476.html

Faster R-CNN安装

http://www.fx114.net/qa-71-160758.aspx
http://www.voidcn.com/blog/a125930123/article/p-6175805.html
http://www.cnblogs.com/zjutzz/p/6034408.html
http://www.cnblogs.com/guanyu-zuike/p/5935691.html
http://www.cnblogs.com/empty16/p/4828476.html

2017.2.26

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值