注:本文为基于微软的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
文件,并显示,如下
参考