ssd-caffe 可视化loss与accuracy曲线

将SSD训练过程中生成的日志文件可视化(以VOC训练为例)

1、默认caffe_root为caffe

在/caffe/jobs/VGGNet/VOC0712/SSD_300x300文件夹中会有VGG_VOC0712_SSD_300x300.log文件。

我们利用这个文件绘制训练过程中的loss与accuracy曲线。

2、使用caffe/tools/extra中自带日志分析工具

caffe/tools/extra路径下执行命令行:

./parse_log.sh ../../jobs/VGGNet/VOC0712/SSD_300x300/VGG_VOC0712_SSD_300x300.log 

将在当前目录下生成VGG_VOC0712_SSD_300x300.log.test,VGG_VOC0712_SSD_300x300.log.train两个文件。

文件内容:

ssd-caffe 可视化loss与accuracy曲线_第1张图片

3、在caffe/tools/extra执行

cp plot_log.gnuplot.example plot_log.gnuplot

 修改代码

gedit plot_log.gnuplot

1、33行,设置图片名称

ssd-caffe 可视化loss与accuracy曲线_第2张图片 

2、44行。画图,更改为自己的文件

ssd-caffe 可视化loss与accuracy曲线_第3张图片

3、caffe/tools/extra路径执行:

gnuplot plot_log.gnuplot

 

你可能感兴趣的:(ssd-caffe 可视化loss与accuracy曲线)