当前位置:   article > 正文

利用caffe工具绘制训练模型中train accuracy和loss曲线

train accuracy

参考博文:(1)https://blog.csdn.net/yx2017/article/details/75045078​​​​​​

                 (2)http://www.cnblogs.com/AliceHu/p/6529383.html

                 (3)https://www.cnblogs.com/wuchaodzxx/p/5904180.html 

                 (4)https://blog.csdn.net/liuweizj12/article/details/64920428 


1. 记录训练日志:

  1. #!/usr/bin/env sh
  2. set -e
  3. echo "begin:"
  4. TOOLS=/home/yangjian/caffe-master/build/tools
  5. LOG=/home/yangjian/caffe-master/myfile_jian/log/log-`date +%Y-%m-%d-%H-%M-%S`.log
  6. $TOOLS/caffe train \
  7. --solver=/home/yangjian/caffe-master/myfile_jian/solver50.prototxt 2>&1 | tee $LOG $@
  8. echo "end"

2. 需要的文件:

  1. 1 caffe/tools/extra/parse_log.sh
  2. 2 caffe/tools/extra/extract_seconds.py
  3. 3 caffe/tools/extra/plot_training_log.py
  • 将这三个文件复制到log目录下,将plot_training_log.py.example重命名为plot_training_log.py
  • 对这三个文件赋予可执行权限:chmod a+x 三个文件

3. 解析训练日志:(此方法该步骤可以不用)

./parse_log.sh log-2018-07-27-10-37-42.log

此时,在log文件下面生成了log-2018-07-27-10-37-42.log.train和log-2018-07-27-10-37-42.log.test文件,里面保存了提取出来的seconds,loss,accuracy信息。

4. 绘制图片:

 ./plot_training_log.py 0 save.png log-2018-07-27-10-37-42.log

其中:

  • 0 为曲线类型
  • save.png为保存图片的文件名
  • log-2018-07-27-10-37-42.log为上述所得日志

5. 参数的含义:

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

参数4和5由于训练日志里最后生成模型时会缺少最后一次显示迭代的学习率而绘图出错,其他参数没有问题 

6. 结果展示

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Monodyee/article/detail/133401?site
推荐阅读
相关标签
  

闽ICP备14008679号