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

1.首先将训练信息保存至log文件

sudo build/tools/caffe train -solver /home/tiankong/Desktop/TSR/final_detec/Final_Training/solver.prototxt 2>&1| tee /home/tiankong/Downloads/caffe/Log/temp.log

1. tee的意思就是命令行信息重定向的命令

2. 2>&1的意思是错误的信息也当成标准信息输出,这样能够保证输出log信息的完整性。


2.解析训练日志

caffe-master/tools/extra/parse_log.sh、caffe-master/tools/extra/extract_seconds.py和caffe-master/tools/extra/plot_training_log.py.example 3个脚本文件拷贝到Log 文件夹下,执行:

1
./parse_log.sh caffe.liuyun-860-088cn.root.log.INFO.20160830-090533.5367

 这样就会在当前文件夹下生成一个.train文件和一个.test文件

3.生成图片

执行

1
./plot_training_log.py.example 6  train_loss.png caffe.liuyun-860-088cn.root.log

注意:一定将caffe.liuyun-860-088cn.root.log.INFO.20160830-090533.5367改为caffe.liuyun-860-088cn.root.log,.log为后缀。

就可以生成训练过程中的Train loss  vs. Iters 曲线,其中6代表曲线类型, train_loss.png 代表保存的图片名称

caffe中支持很多种曲线绘制,通过指定不同的类型参数即可,具体参数如下:

1
2
3
4
5
6
7
8
9
10
11
12
Notes: 
     1. Supporting multiple logs. 
     2. Log file name must end with the lower-cased ".log"
Supported chart types: 
     0: Test accuracy  vs. Iters 
     1: Test accuracy  vs. Seconds 
     2: Test loss  vs. Iters 
     3: Test loss  vs. Seconds 
     4: Train learning rate  vs. Iters 
     5: Train learning rate  vs. Seconds 
     6: Train loss  vs. Iters 
     7: Train loss  vs. Seconds 

 最后,看一下效果:


猜你喜欢

转载自blog.csdn.net/tiankongtiankong01/article/details/80383580