1. 程式人生 > >用caffe自帶工具plot_training_log.py繪製loss和accuracy曲線

用caffe自帶工具plot_training_log.py繪製loss和accuracy曲線

將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