您的位置：首页 > Web前端

pycaffe 配置

2016-03-25 16:21 369 查看

本系列文章由 @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/
$ make clean
$ make all -j8
$ make test -j8
$ make runtest -j8
$ 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

内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理

标签： caffe pycaffe

相关文章推荐

新的分享

章节导航