最近做实验用到MatCaffe,本文就简单总结以下MatCaffe的用法。还没安装的赶快看官网教程安装吧。MatCaffe的用法与PyCaffe挺类似的。
model ='./models/bvlc_reference_caffenet/deploy.prototxt';
weights='./models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel';
上面是设置好的你的prototxt文件路径(网络结构文件)以及caffemodel文件路径(模型文件)。
设置mode和device
Mode和device应该在你创建net或者solver之前设置。
用CPU或者GPU:
caffe.set_mode_cpu();
Use GPU and specify its gpu_id:
caffe.set_mode_gpu();
caffe.set_device(gpu_id);
创建网络:
net = caffe.Net(model, weights, 'test');
% create net and load weights
或者可以后来载入weights:
net = caffe.Net(model, 'test');
% create net but not load weights
net.copy_from(weights);
% load weights
网络的具体结构是下面的样子:
layer_vec: [1x23 caffe.Layer]