caffe-ssd中使用gnuplot绘制loss-acc曲线(

2020-04-11  本文已影响0人  几夏经秋

1.首先检查SSD训练的完整log日志,因为所需要的东西都在日志里面。日志目录在:jobs里面,名字为VGG_MyDataSet_SSD_300x300.log。

利用此log文件生成相应大的.test和.train文件,具体命令如下:(当前目录为caffe-ssd,输入上面的命令)

cd tools/extra

./parse_log.sh ../../jobs/VGGNet/MyDataSet/SSD_300x300/VGG_MyDataSet_SSD_300x300.log

parse_log.sh是caffe自带的log分析工具,一般在caffe的tools文件内的extra文件夹中。

2.安装gnuplot工具。

apt-get install gnuplot

安装完成之后将caffe-ssd下的tools/extra中的新建py_mAP_loss.py文件

文件中代码如下:

reset

set terminal png

set output "VGG_MyDataSet_SSD_300x300.png"

set style data lines

set key right

set term pngcairo size 1200,800  #这里是设置图片的大小

set multiplot

set title "SSD Training loss and mAP vs. training iterations"

set xlabel "Training iterations"

set ylabel "Training loss"

set y2label "mAP"

set y2tics

set ytics nomirror

# using 1:3 表示使用log-data.train的第一和第三列分别为x轴y轴

plot "VGG_MyDataSet_SSD_300x300.log.train" using 1:3 w l axis x1y1 title "train loss", "VGG_MyDataSet_SSD_300x300.log.test" using 1:3  w l axis x1y2 title "mAP"

之后命令行输入

 gnuplot py_mAP_loss.py 

大功告成:

对于这个工具的使用还需要进一步的学习。

上一篇下一篇

猜你喜欢

热点阅读