在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'

网友评论