caffe-ssd中使用gnuplot繪制loss-acc曲線(雙Y軸)

1.首先檢查SSD訓(xùn)練的完整log日志,因為所需要的東西都在日志里面。日志目錄在:jobs里面,名字為VGG_MyDataSet_SSD_300x300.log。

利用此log文件生成相應(yīng)大的.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文件內(nèi)的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? #這里是設(shè)置圖片的大小

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?

大功告成:

對于這個工具的使用還需要進一步的學(xué)習。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容