pycaffe 配置
本系列文章由 @yhl_leo 出品,转载请注明出处。
文章链接: http://blog.csdn.net/yhl_leo/article/details/50967820
本文将继续接着上一篇博客(Ubuntu14.04 安装CUDA7.5 + Caffe + cuDNN),讲述如何配置pycaffe
,以及运行rbgirshick提供的py-faster-cnn
代码,先上图:
1 基本依赖项
首先安装一些依赖项:
$ sudo apt-get install python-skimage python-protobuf python-yaml
$ sudo pip install easydict
2 OpenCV
在编译过程中使用前文的OpenCV编译方式,后面在编译时出现了链接opencv的bug,推荐使用下面的方法:Ubuntu 15.10 OpenCV 3.1 Installation Guide
opencv下载链接:http://sourceforge.net/projects/opencvlibrary/files/opencv-unix/ ,本人下载的是3.1.0版本。
下载完成后,解压并创建build
文件夹:
$ mkdir build
$ cd build/
$ cmake -D CMAKE_BUILD_TYPE=RELEASE -D CMAKE_INSTALL_PREFIX=/usr/local -D WITH_TBB=ON -D WITH_V4L=ON -D WITH_QT=ON -D WITH_OPENGL=ON ..
$ make
安装:
$ sudo make install
$ sudo /bin/bash -c 'echo "/usr/local/lib" > /etc/ld.so.conf.d/opencv.conf'
$ sudo ldconfig
$ sudo apt-get update
3 环境变量
编译完成后,需要添一些环境变量:
$ export PYTHONPATH=${HOME}/caffe/python:$PYTHONPATH
$ export MATLABPATH=${HOME}/caffe/matlab:$MATLABPATH
$ export LD_LIBRARY_PATH=${HOME}/caffe/build/lib:$LD_LIBRARY_PATH
$ export PATH=/usr/local/cuda-7.5/bin:$PATH
$ export LD_LIBRARY_PATH=/usr/local/lib/:$LD_LIBRARY_PATH
(其中第二行是配置matcaffe
的,如果没有安装matlab就不要输入。)
不要直接在终端里输入,看到一篇帖子说,这种在终端里输入的方式,是暂时性的,当你关闭输入的那个终端窗口或者reboot
后,设置的环境变量就没了,所以,建议把这些环境变量添加到/etc/profile
文件的末尾:
$ sudo vi /etc/profile
添加完成后,输入
source /etc/profile
建议重启(reboot
),然后检查一下环境变量:
$ echo $PYTHONPATH
4 修改Makefile.config
提供本人的Makefile.config
的文件:yhlleo/caffe.Makefile.config
如果没有安装matlab,请把MATLAB_LIB
保持注释状态。
matlab 2014a的安装方法可以参照:Ubuntu14.04安装Matlab2014a
5 编译pycaffe
$ cd ~/caffe/
$ sudo make clean
$ sudo make all -j8
$ sudo make test -j8
$ sudo make runtest -j8
$ sudo make pycaffe -j8
如果编译出现make: Nothing to be done for 'pycaffe'
,就使用命令sudo make clean
。
编译有没有问题最简单的测试方法,就是在终端中依次输入:
$ python
>>>import caffe
如果提示没找到caffe
模块,就说明没安装好,仔细检查是哪里出问题。
6 测试py-faster-rcnn
下载rbgirshick提供的开源代码:rbgirshick/py-faster-rcnn
其中的README.md,对于如何下载代码、数据以及运行测试demo讲述的比较清晰,就不累述。
下载数据,运行./tools/demo.py
后,就可以得到一组检测结果图片(开篇给出的图片,是在sublime中编译执行的结果)。
祝好运~~
源自本人博客网站:yhl’s blog