Windows玩转Caffe（五）： mnist模型可视化探知，Matlab绘制其Accuracy和Loss曲线

欲练此功，必先自攻Windows玩转Caffe（三）：由图片训练mnist模型，并用其模型识别手写数字【多图出品】http://blog.youkuaiyun.com/u012958854/article/details/78193551

先上动态效果图：

 

 

1.      下载Matlab，并安装

本人安装的Matlab2015b，下载地址：http://jingyan.baidu.com/article/eae07827a466a31fed548563.html

 

2.      修改配置，编译matcaffe

打开CommonSettings.props，编辑要点：

（详见http://blog.youkuaiyun.com/u012958854/article/details/78123610）

编译matcaffe工程，在Release下将生成matcaffe文件夹。

 

3．编辑.m文件，代码如下：

clear 

clc 

close all 

format long %设置精度，caffe的损失貌似精度在小数点后面好几位 

addpath('..\..\Build\x64\Release\matcaffe') 

addpath('..\..')

caffe.reset_all%重设网络，否则载入两个网络会卡住 

solver=caffe.Solver('examples\mnist\lenet_solver.prototxt');%载入网络 

loss=[];%记录相邻两个loss 

accuracy=[];%记录相邻两个accuracy 

hold on%画图用的 

accuracy_init=0; 

loss_init=0; 

for i=1:100 %10000

   solver.step(1);%每迭代一次就取一次loss和accuracy 

   iter=solver.iter(); 

   loss=solver.net.blobs('loss').get_data();%取训练集的loss 

   accuracy=solver.test_nets.blobs('accuracy').get_data();%取验证集的accuracy 

     

    %画loss折线图 

   x=[i-1,i]; 

   y=[loss_init loss]; 

   z=[accuracy_init accuracy];

   plot(x,y,x,z,'r-') 

   %plot(x,z,'r-')

   drawnow 

   loss_init=loss; 

   accuracy_init=accuracy;

end

grid on

title('mnist折线图');

xlabel('iter');

ylabel('accuracy & loss');

text(90,0.5,'loss')    %将cosx这个注解加到坐标中的某个位置

text(5, 0.2, 'accuracy')%gtext('accuracy')      % 用鼠标的光标定位,将sinx这个注解放在你鼠标点击的地方

%axis([xmin xmax ymin ymax]) 设置坐标轴的最小最大值

       保存文件至caffe-master\matlab\demo目录下。

 

4．最后一步，在Matlab中运行

     打开Matlab打开.m文件，把当前文件夹设置为caffe-master根目录（否则会报找不到文件错误甚至导致Matlab崩溃！！！）；

       点击运行，并选择保存路径（注意不是第一项更改文件夹），最终效果图：

（运行过程中可以看到Matlab弹出的如篇头所见的动态效果图，想要保存为gif图片的话，方法请见：http://download.youkuaiyun.com/download/u012958854/10027877）