Caffe學習筆記(4) -- 視覺化訓練結果
阿新 • • 發佈:2019-01-04
注:本文為基於微軟的Windows Caffe
Caffe提供了一個視覺化訓練結果的工具,下面就是具體的操作方法
caffe的根目錄下文統一定義為CAFFE_ROOT
首先,預設情況下執行caffe.exe訓練時是不生成日誌的。我們要通過日誌來解析訓練結果
1. 在CAFFE_ROOT
下新建一個run_mnist.sh
檔案,內容如下
LOG=./log/train-`date +%Y-%m-%d-%H-%M-%S`.log
CAFFE=E:/TSY/Code/caffe-Microsoft/Build/x64/Release
$CAFFE/caffe train --solver=examples/mnist/lenet_solver.prototxt --gpu=0 2>&1 | tee $LOG
2. 執行
CAFFE_ROOT\run_mnist.sh
,生成的log檔案會儲存在CAFFE_ROOT\log\
下面 3. 開啟
CAFFE_ROOT\tools\extra
,在該目錄下新建一個文字文件,命名為draw_result.sh
,文件內容如下
python plot_training_log.py 0 ./result.png E:/TSY/Code/caffe-Microsoft/log/train-2016-06-16-06-20-36.log
4. 將
CAFFE_ROOT\tools\extra\plot_training_log.py.example
plot_training_log.py
5. 使用shell執行
draw_result.sh
,我這裡使用的是cgyshell執行。執行成功後會在當前目錄生成一個result.png
檔案,並顯示,如下 參考