Windows玩转Caffe(五): mnist模型可视化探知,Matlab绘制其Accuracy和Loss曲线

欲练此功,必先自攻Windows玩转Caffe(三):由图片训练mnist模型,并用其模型识别手写数字【多图出品】http://blog.csdn.net/u012958854/article/details/78193551

先上动态效果图:

 Windows玩转Caffe(五): mnist模型可视化探知,Matlab绘制其Accuracy和Loss曲线

 

1.      下载Matlab,并安装

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

 

2.      修改配置,编译matcaffe

打开CommonSettings.props,编辑要点:

Windows玩转Caffe(五): mnist模型可视化探知,Matlab绘制其Accuracy和Loss曲线

(详见http://blog.csdn.net/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崩溃!!!);

       点击运行,并选择保存路径(注意不是第一项更改文件夹),最终效果图:

Windows玩转Caffe(五): mnist模型可视化探知,Matlab绘制其Accuracy和Loss曲线

(运行过程中可以看到Matlab弹出的如篇头所见的动态效果图,想要保存为gif图片的话,方法请见:http://download.csdn.net/download/u012958854/10027877)