利用caffe中自帶的工具來視覺化loss 和accuracy
阿新 • • 發佈:2019-01-02
以前只是一股腦的訓練,卻很少注意到這些,今天仔細研究了下,發現caffe自帶技能包.方法如下:
1訓練,和以前略有不同的是,./XX.sh|& tee xx.log,保證在caffe-master目錄下生成日誌檔案,或者去根目錄下的temp中尋找也可以.
2在caffe-master下執行./tools/extra/parse_log.sh fexpress.log,在當前路徑下生成fexpress.log.train和fexpress.log.test兩個檔案.將這兩個檔案複製到/tools/extra/目錄下.
3確保安裝了gnuplot. sudo apt-get install gnuplot
4 開啟plot_log.gnuplot.example,可以將your_chart_name.gnuplot plot_log.gnuplot.example
注意:
plot_log.gnuplot.example
檔案中前半部分顯示train的內容,後半部分顯示test的內容.
# Training loss vs. training iterations set title "Training loss vs. training iterations" set xlabel "Training iterations" set ylabel "Training loss" plot "finetune_age.log.train" using 1:3 title "age" #mnist.log.train改成自己的檔名
5 為了顯示測試的準確率,在plot_log.gnuplot.example的末尾新增如下命令列:
# Test accuracy vs. training time
set ylabel "Test accuracy"
set xlabel "Training time"
plot "fexpress.log.test" using 1:2 title "express"
有人進一步註釋掉41-44行(把#在該行的開頭)。 當同樣執行下面的命令和更早版本,下面your_chart_name.png輸出。
執行:
gnuplot plot_log.gnuplot.example
參考連結: