matlab - 使用 caffe 提取过滤器和偏差

Question

.mat我想从我自己的caffemodel中提取过滤器和偏差，（无需可视化该功能）并且我想将它们保存到MATLAB. 我曾经matcaffe做过这个问题：

我的解决方法：

addpath('/home/jensen810814/code/caffe/matlab/caffe/');
addpath('/home/jensen810814/code/caffe/examples/super_resolution/');
addpath('/home/jensen810814/code/caffe/examples/super_resolution/Model/');
caffe('init','deploy.prototxt','super_resolution_iter_1000.caffemodel');
caffe('forward',{});
weights = caffe('get_weights');
save('/home/jensen810814/code/caffe/examples/super_resolution/filter_data.mat','weights');

但是发生了一些错误：

Error using caffe
Expected 3 arguments, got 2
Error in model (line 4)
caffe('init','supe_resolution_train_test.prototxt','super_resolution_iter_1000.caffemodel');

我不知道如何解决这个问题。谁能帮我？请和谢谢。

Answer 1

'init'您需要为阶段名称提供额外的参数。
我相信你的情况

 caffe('init',...
       'supe_resolution_train_test.prototxt',...
       'super_resolution_iter_1000.caffemodel',...
       'test');