caffe的Matlab接口的使用方法

最新推荐文章于 2020-11-16 17:23:21 发布

及时澍雨Timely

最新推荐文章于 2020-11-16 17:23:21 发布

阅读量5.6w

点赞数 19

分类专栏： Deep Learning 文章标签： caffe matlab matcaffe 接口 solver

本文链接：https://blog.csdn.net/ws_20100/article/details/50525879

版权

本文介绍了如何编译和使用MatCaffe，包括设置模式和设备、创建网络、访问layers和blobs、前向后向计算、reshape、训练网络以及输入输出的操作。详细阐述了在Matlab环境中如何进行Caffe模型的交互和训练过程。

摘要由CSDN通过智能技术生成

编译MatCaffe

使用如下命令编译MatCaffe

make all matcaffe

之后，你可以用以下命令测试MatCaffe：

make mattest

如果你在运行上面命令时，遇到如下错误：libstdc++.so.6 version ‘GLIBCXX_3.4.15’ not found，说明你的Matlab库不匹配。你需要在启动Matlab之前运行如下命令：

export LD_LIBRARY_PATH=/opt/intel/mkl/lib/intel64:/usr/local/cuda/lib64
export LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libstdc++.so.6

在Caffe根目录启动Matlab之后需要增加路径：

addpath ./matlab

你可以使用savepath来保存Matlab搜索路径，这样下次就不用再添加路径了。

MatCaffe 和 PyCaffe 的使用方法很相似。

下面将用一个例子来解释MatCaffe的具体使用细节，假设你已经下载了BVLC CaffeNet，并且在caffe根目录启动matlab。

model = './models/bvlc_reference_caffenet/deploy.prototxt';
weights = './models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel';

模式和设备的设置必须在创建一个net或solver之前。

使用CPU：

caffe.set_mode_cpu();

使用GPU并指定gpu_id：

caffe.set_mode_gpu();
caffe.set_device(gpu_id);

创建一个网络：

net = caffe.Net(model, weights, 'test'); % create net and load weights

或者

net = caffe.Net(model, 'test'); % create net but not load weights
net.copy_from(weights); % load weights

它可以创建一个如下的net对象：

关注

专栏目录