使用caffe提取过滤器和偏差

时间:2015-04-16 08:51:55

标签: matlab machine-learning neural-network deep-learning caffe

我想从我自己的caffemodel中提取过滤器和偏差,(无需可视化该功能),我想将它们保存到.mat的{​​{1}}文件中。 我使用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');

我不知道如何解决这个问题。 谁能帮我?拜托,谢谢。

1 个答案:

答案 0 :(得分:0)

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

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