1. 程式人生 > >利用caffe中自帶的工具來視覺化loss 和accuracy

利用caffe中自帶的工具來視覺化loss 和accuracy

以前只是一股腦的訓練,卻很少注意到這些,今天仔細研究了下,發現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.
png修改成你要儲存的圖片名.其中x軸表示訓練的迭代次數,y軸表示訓練的loss.然後執行:
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

參考連結:
http://iamrobotandproud.hatenablog.com/entry/2015/03/16/105746
,非常感謝!