Caffe学习笔记（4） -- 可视化训练结果

注：本文为基于微软的Windows Caffe

Caffe提供了一个可视化训练结果的工具，下面就是具体的操作方法

caffe的根目录下文统一定义为

CAFFE_ROOT

首先，默认情况下执行caffe.exe训练时是不生成日志的。我们要通过日志来解析训练结果

1. 在

CAFFE_ROOT

下新建一个

run_mnist.sh

文件，内容如下

LOG=./log/train-`date +%Y-%m-%d-%H-%M-%S`.log

CAFFE=E:/TSY/Code/caffe-Microsoft/Build/x64/Release

$CAFFE/caffe train --solver=examples/mnist/lenet_solver.prototxt --gpu=0 2>&1 | tee $LOG

2. 运行

CAFFE_ROOT\run_mnist.sh

，生成的log文件会存储在

CAFFE_ROOT\log\

下面



3. 打开

CAFFE_ROOT\tools\extra

，在该目录下新建一个文本文档，命名为

draw_result.sh

，文档内容如下

python plot_training_log.py 0 ./result.png E:/TSY/Code/caffe-Microsoft/log/train-2016-06-16-06-20-36.log

4. 将

CAFFE_ROOT\tools\extra\plot_training_log.py.example

更名为

plot_training_log.py

5. 使用shell运行

draw_result.sh

,我这里使用的是cgyshell运行。运行成功后会在当前目录生成一个

result.png

文件，并显示，如下



参考

https://www.zhihu.com/question/36652304