Windows玩转Caffe(五): mnist模型可视化探知,Matlab绘制其Accuracy和Loss曲线
欲练此功,必先自攻Windows玩转Caffe(三):由图片训练mnist模型,并用其模型识别手写数字【多图出品】http://blog.csdn.net/u012958854/article/details/78193551
先上动态效果图:
1. 下载Matlab,并安装
本人安装的Matlab2015b,下载地址:http://jingyan.baidu.com/article/eae07827a466a31fed548563.html
2. 修改配置,编译matcaffe
打开CommonSettings.props,编辑要点:
(详见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崩溃!!!);
点击运行,并选择保存路径(注意不是第一项更改文件夹),最终效果图:
(运行过程中可以看到Matlab弹出的如篇头所见的动态效果图,想要保存为gif图片的话,方法请见:http://download.csdn.net/download/u012958854/10027877)