对训练的caffe.bat文件进行编辑,在最后加上>trainlog.txt 2>&1,即可实现将cmd打印出来的信息保存到trainlog.txt文件中。
SET GLOG_logtostderr=1
caffe.exe路径 train --solver solver.prototxt路径 >trainlog.txt 2>&1
pause
这样在训练完成之后,日志信息全都保存到了trainlog.txt文件中。
然后进行绘图。
代码以及部分注释如下所示。
%%loss示例
clc;clear
logName='C:\Users\dell\Desktop\trainlog.txt';%%trainlog.txt的路径
dir_log='C:\Users\dell\Desktop\loss\';%%保存以下五个txt文件的路径
fid = fopen(logName, 'r'); %%read 读取日志文件
test_loss = fopen([dir_log 'test_loss.txt'], 'w'); %%write 往txt文件中写内容
train_loss = fopen([dir_log 'train_loss.txt'], 'w');
train_lr = fopen([dir_log 'train_lr