本文主要介绍,将caffe训练得到的accracy,loss进行图像化。
对于一般caffe训练结果的可视化:
1.在训练时,需要将训练的结果保存日志。
train.sh:
#!/usr/bin/env sh
TOOLS=/home/zhuangni/code/Multi-Task/caffe-master/build/tools
GLOG_log_dir='/home/zhuangni/code/Multi-Task/experiment_single/attr1/vgg/log/' \
$TOOLS/caffe train \
--solver=/home/zhuangni/code/Multi-Task/experiment_single/attr1/vgg/solver.prototxt \
--weights=/home/zhuangni/code/Multi-Task/experiment_single/attr1/vgg/face_snapshot_iter_450000.caffemodel \
--gpu=0
GLOG_log_dir为日志保存路径。
日志名自动生成为: caffe.zhuangni.zhuangni.log.INFO.20161020-215304.3679
log文件夹下自动生成: caffe.INFO 和 caffe.zhuangni.zhuangni.log.INFO.20161020-215304.3679 文件。其中 caffe.zhuangni.zhuangni.log.INFO.20161020-215304.3679 文件即为日志文件。
2.新建一个acc目录,
1)将日志文件caffe.zhuangni.zhuangni.log.INFO.20161020-215304.3679拷入,并更改为 my.log
2)将caffe-master/tools/extra目录里的extract_seconds.py , plot_training_log.py.example, parse_log.sh三个文件拷入
3.运行
./plot_training_log.py.example 0 demo.png my.log
运行参数:
=======&#