以前只是一股脑的训练,却很少注意到这些,今天仔细研究了下,发现caffe自带技能包.方法如下:
1训练,和以前略有不同的是,./XX.sh|& tee xx.log,保证在caffe-master目录下生成日志文件,或者去根目录下的temp中寻找也可以.
2在caffe-master下运行./tools/extra/parse_log.sh fexpress.log,在当前路径下生成fexpress.log.train和fexpress.log.test两个文件.将这两个文件复制到/tools/extra/目录下.
3确保安装了gnuplot. sudo apt-get install gnuplot
4 打开plot_log.gnuplot.example,可以将your_chart_name.png修改成你要保存的图片名.其中x轴表示训练的迭代次数,y轴表示训练的loss.然后运行:
gnuplot plot_log.gnuplot.example
注意:
plot_log.gnuplot.example
文件中前半部分显示train的内容,后半部分显示test的内容.
# Training loss vs. training iterations
set title "Training loss vs. training iterations"
set xlabel "Training iterations"
set ylabel "Training loss"
plot "finetune_age.log.train" using 1:3 title "age"
#mnist.log.train改成自己的文件名
5 为了显示测试的准确率,在plot_log.gnuplot.example的末尾添加如下命令行:
# Test accuracy vs. training time
set ylabel "Test accuracy"
set xlabel "Training time"
plot "fexpress.log.test" using 1:2 title "express"
有人进一步注释掉41-44行(把#在该行的开头)。 当同样运行下面的命令和更早版本,下面your_chart_name.png输出。
运行:
gnuplot plot_log.gnuplot.example
参考链接: http://iamrobotandproud.hatenablog.com/entry/2015/03/16/105746,非常感谢!