1.训练模型并获取训练日志
用CMD调用Caffe进行模型训练并将日志重定向到txt文件,命令格式如下:
#命令格式:
command 2> logfile_path & type logfile_path
#示例:
Build\x64\Release\caffe.exe train --solver=examples\mnist\lenet_solver_adam.prototxt 2> examples\mnist\train_log.txt & type examples\mnist\train_log.txt
#回车执行命令,训练完成后,CMD窗口的训练日志会被保存到examples\mnist\train_log.txt中。
2.解析日志文件并绘制loss和accuracy曲线
将上一步得到的日志文件train_log.txt的完整路径输入下面Matlab脚本定义的PlotLossAndAccuracy()函数,即可解析并绘制loss和accuracy曲线
function result=PlotLossAndAccuracy(kLogName)
%输入caffe模型训练日志的路径,绘制loss和accuracy曲线
%返回值result=[accuracy,testloss,trainloss]
%示例:
%PlotLossAndAccuracy('E:\caffe-window\新建文本文档.txt')
%初始化变量
ind=strfind(kLogName,'\');%找到所有'\'
ind1=ind(end)+1;
ind=strfind(kLogName,'_train_log');