deep-learning - 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'));