caffe-Matlab接口(matcaffe)能夠讓妳把caffe框架整合到妳的matlab代碼中。
1、構建MatCaffe 接口
使用make all matcaffe,如果成功,可以使用make mattest測試。
2、使用MatCaffe 。
在運用MatCaffe 創建net或者solver之前,必須先設置運行模式和設備:
使用cpu:
caffe.set_mode_cpu();
使用gpu和指定gpu_id:
caffe.set_mode_gpu();
caffe.set_device(gpu_id);3、創建網絡並且訪問它的layer層和blob數據
net=caffe.Net(model,weight,'test');創建網絡並載入weights
或者可以這樣定義:
net=caffe.Net(model,'test');%創建網絡但不載入數據
net.copy_from(weights);%載入數據
4、訪問數據並更改數據,這裏我們壹個例子為例,
假設我們要獲取到blob的數據data,並將data*10再存回blob.我們可以這樣做:
net.blobs('data').set_data(net.blobs('data').get_data()*10);
5、保存妳要修改的網絡:
net.save('my_net.caffemodel');
6、獲取壹個層layer的類型type:
layer_type=net.layers('convl').type;