1. 程式人生 > >Caffe學習筆記(4) -- 視覺化訓練結果

Caffe學習筆記(4) -- 視覺化訓練結果

注:本文為基於微軟的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 檔案,並顯示,如下
這裡寫圖片描述

參考