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