caffe命令列訓練以及測試

2021-08-16 21:22:28 字數 1493 閱讀 4743

網易雲課堂(程式中檔案放在目錄~/documents/caffe/projects/mnist中):

在終端中輸入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的輸出,修改後執行成功

caffe命令列解析

caffe提供三種介面,一般是c python matlab。一般可執行檔案都是放在 build tools 資料夾內,在命令列執行命令必須現在該目錄下。也有可能不在說不定在別的地方,找caffe.exe就對了 caffe的命令形式如下 caffe 其中的command有這樣四種 其中的args引數...

Caffe 命令列解析

sudo sh build tools caffe train solver examples mnist train lenet.shcaffe程式命令列執行格式如下 caffe 其中有這樣四種 build tools caffe train solver examples mnist lenet...

Caffe學習 命令列解析

caffe的執行提供三種介面 c 介面 命令列 python介面和matlab介面。本文先對命令列進行解析,後續會依次介紹其它兩個介面。caffe的c 主程式 caffe.cpp 放在根目錄下的tools資料夾內,當然還有一些其它的功能檔案,如 convert imageset.cpp,train ...