caffe 训练增加日志,画accuracy曲线

2019-11-08 02:05:30

字体：大中小

来源：转载

供稿：网友

首先在当前创建文件夹log : mkdir log 在bash文件 train.sh 添加：

LOG=log/train-`date +%Y-%m-%d-%H-%M-%S`.logTOOLS=/opt/caffe-py3/build/tools$TOOLS/caffe train -solver='LCNN_solver.PRototxt' --gpu=0 2>&1 | tee $LOG

训练玩在log文件夹下会有 .log文件

把caffe-master/tools/extra/路径下的3个文件 - parse_log.sh - extract_seconds.py - plot_training_log.py.example 执行bash parse_log.sh ***.log,会生成.train 和.test文件再执行 ./plot_training_log.py.example 6 trainloss.png train-2017-02-17-10-57-03.log就会得到图，其中参数如下：

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

我执行的是6 trainloss vs. iters