用caffe自带工具plot_training_log.py绘制loss和accuracy曲线

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/zhongqianli/article/details/85602643

将caffe/tools/extra/plot_training_log.py.example脚本重命名为plot_training_log.py（这个脚本依赖所在目录下的其他文件，使用时要确保它们和plot_training_log.py在同一目录），然后编写shell脚本，用于绘制loss、accuracy、lr变化曲线：

#!/usr/bin/env sh
set -e

#TRAIN_LOG_NAME="cifar10_resnet20_train.log"
TRAIN_LOG_NAME="cifar10_resnet56_train.log"

python2 plot_training_log.py 0 ${TRAIN_LOG_NAME}.acc.png $TRAIN_LOG_NAME
python2 plot_training_log.py 4 ${TRAIN_LOG_NAME}.lr.png $TRAIN_LOG_NAME
python2 plot_training_log.py 6 ${TRAIN_LOG_NAME}.loss.png $TRAIN_LOG_NAME