deep-learning - caffe 如何使用matlab设置过滤器？

Question

我想在 Caffe 中使用 matlab 更新卷积滤波器。我该怎么做？到目前为止，我的代码是：

caffe.reset_all(); % reset caffe
caffe.set_mode_gpu();
gpu_id = 0;  % we will use the first gpu in this demo
caffe.set_device(gpu_id);

net_model = [model_dir, 'train_images.prototxt'];
net = caffe.Net(net_model, 'train');

net.blobs('conv1').set_data([1,1,0]);

Answer 1

以下工作正常。

caffe.reset_all(); % reset caffe
caffe.set_mode_gpu();  
gpu_id = 0;  % we will use the first gpu in this demo    
caffe.set_device(gpu_id);
net_model = ['net_images.prototxt'];
net = caffe.Net(net_model, 'train')
a = zeros(1,1,3,2);
a(1,1,:,:) = [[1,0,0];[0,1,0]]'; % caffe uses BGR color channel order
         net.layers('select_B_G').params(1).set_data(a);
solver = caffe.Solver(solverFN);
solver.solve();
net.save(fullfile(model_dir, 'my_net.caffemodel'));