Windows下编译Caffe并编译Matlab接口

来源:互联网 发布:网络大电影监禁风暴 编辑:程序博客网 时间:2024/06/11 06:34

一直嫌麻烦没去装caffe,近日抽时间装好了Caffe,并且编译好了C++,Matlab的接口,现在把安装过程记录如下:

一,安装环境

Win7 64位

VS2012

Matlab R2012b

CUDA 6.5

二.安装步骤

1.Caffe 源程序

由于我是在windows环境下安装Caffe,所以是无法直接用原版的Caffe的。我是从CSDN上下载了已经预装好诸多第三方库的一个Caffe版本(记为Caffe0),但是这个版本没有Matlab的编译接口,因此又从github上找了caffe-windows(记为Caffe1)的版本,这两个版本的Caffe工程我已经放到百度网盘上面去了。

http://pan.baidu.com/s/1gdpnXb9


2.编译

打开Caffe1/build,把MSVCmex这个文件夹拷贝到caffe0/build中。

用ultraedit打开MSVCmex文件夹里面的vc++工程,将cuda和matlab的目录改成自己电脑里面的目录,否则用VS2012会打不开工程。

用VS2012打开工程,选择右键caffe工程->属性->配置属性->右上角的配置管理器,然后将平台由win32改成X64

接下来就是编译,会需要一些时间,编译通过之后就会在caffe/matlab/caffe目录里面生成caffe.mex等文件,这里的caffe.mex就相当于c++里面的dll(lib)库,只是它是面向matlab的而已。


3.Matlab Demo

首先要准备imagenet的Caffe模型,在网上可以下载到,我放在了百度网盘里面:

http://pan.baidu.com/s/1ntjCcXV


打开matlab 2012 ,进入caffe/matlab/caffe目录,然后依次在工作空间输入以下指令:

im = imread('peppers.png');

[scores, maxlabel] = matcaffe_demo(im,1);

一般情况下就能运行通过了,最后max_label是946。到这里就说明caffe的matlab接口已经配置完毕。

1 0