caffe绘制训练过程的loss和accuracy曲线

在caffe的训练过程中,大家难免想图形化自己的训练数据,以便更好的展示结果。如果自己写代码记录训练过程的数据,那就太麻烦了,caffe中其实已经自带了这样的小工具 caffe/tools/extra/plot_training_log.py.example ,使用方法如下:

1.记录训练日志

在训练过程中的命令中加入一行参数 ,实现Log日志的记录。

sh train_lenet.sh >& 1.log

2.绘制

./plot_training_log.py.example   0  1.png       '/home/xuxing/caffe/tools/mnist_lenet.log'



caffe绘制训练过程的loss和accuracy曲线