caffe之學習曲線可視化

一、log文件的獲取

默認，我們在終端上輸入模型訓練指令，訓練的log信息會輸出在標準輸出也即屏幕上，我們要想查看某一步的信息需要滾動滑動條來查看，在終端中查看。可通過下面的方式來將log信息以文件的形式保存到指定位置。

自己寫一個腳本文件，把訓練的各種參數都加在裏面。

#!/usr/bin/env sh
set -e
TOOLS=./build/tools

$TOOLS/caffe train --solver=/home/zht/20170622_workspace/dalunwenfuben2/proto/solver.prototxt $@

其中 $@指的是該腳本在使用的時候可能追加的內容。我這裏將腳本的名字命名爲quick_train.sh(這個名字可以自己定)

將該腳本文件複製到自己caffe的根目錄下。爲當前用戶添加腳本的執行權限

sudo chmod u+x quick_train.sh

執行該腳本文件

./quick_train.sh >& dalunwenfuben2_back.log &

其中>& 是將log信息重定向至dalunwenfuben2_back.log日誌文件內。

&是將腳本後臺運行的意思

二、學習曲線的繪製

caffe本身提供了學習曲線的繪製程序：

./tools/extra/plot_training_log.py.example 0 learn_curve1.png ./dalunwenfuben2_back.log 

在終端執行上述指令可以得到如下的學習曲線。

其中指令中的0是可變選項，還可以選擇1，2。分別有如下對應：

./tools/extra/plot_training_log.py.example 1 learn_curve2.png ./dalunwenfuben2_back.log 

./tools/extra/plot_training_log.py.example 2 learn_curve3.png ./dalunwenfuben2_back.log

圖片上傳功能被限制了，有空再上傳