1. 程式人生 > >caffe命令列訓練以及測試

caffe命令列訓練以及測試

網易雲課堂(程式中檔案放在目錄~/Documents/caffe/projects/mnist中):https://study.163.com/course/courseLearn.htm?courseId=1003491001#/learn/video?lessonId=1004081127&courseId=1003491001

在終端中輸入sudo /home/caffe/build/tools/caffe

會出現許多幫助資訊

caffe後面可以跟4個命令:train、test、device_query、time

1、time:評估模型執行的時間大概是多少

命令(跑CPU):sudo /home/caffe/build/tools/caffe time -model hbk_mnist.prototxt -iterations 10

指定模型以及迭代次數

命令(跑GPU):sudo /home/caffe/build/tools/caffe time -model hbk_mnist.prototxt -iterations 10 -gpu 0

2、train:訓練網路

命令:sudo /home/caffe/build/tools/caffe train -solver hbk_mnist_solver.prototxt

出現錯誤:Unknown bottom blob 'label' (layer 'accuracy', bottom index 1)

將最後兩層,即accuracy和cost層註釋掉即可(問題:不註釋怎麼解決?)

將所有log都輸出到一個檔案中(螢幕沒有輸出):sudo /home/caffe/build/tools/caffe train -solver hbk_mnist_solver.prototxt 2>1 | tee a.log

!!將所有log都輸出到一個檔案中(螢幕既有輸出,檔案也有輸出):sudo /home/caffe/build/tools/caffe train -solver hbk_mnist_solver.prototxt 2>&1 | tee a.log

通過caffe的工具把log中有用的資訊提取出來,使用caffe/tools/extra/parse_log.py

另外一個工具更加直觀的,將log輸出圖,使用caffe/tools/extra/plot_training_log.py.example

輸出幫助資訊:

使用方法:sudo python /home/caffe/tools/extra/plot_training_log.py.example 4 plotlog.png a.log

生成了一張圖片plotlog.png,還有a.log.test和a.log.train兩個檔案

也可以用parse_log.py來生成a.log.test和a.log.train兩個檔案,然後自己進行畫圖

稍微有點問題

=====================分割線4-19======================

訓練時出現的問題,因為在定義網路的檔案中,即hbk_mnist.prototxt檔案中,在測試階段的資料層沒有label的輸出,修改後執行成功