caffe提供三種介面:c++(命令列), python, matlab。如何使用常用的命令呢?
caffe的c++主程式caffe.cpp放在~/caffe/tools/資料夾內,在編譯caffe時生成了可執行檔案在~/caffe/build/tools/中,當然還有一些其它的功能檔案,如:convert_imageset.cpp, train_net.cpp, test_net.cpp等也放在這個資料夾內。
caffe
其中command有四種:
arg引數有:
下面詳細介紹不同引數代表的意義
(1)-solver 指定模型的配置檔案,例如
./build/tools/caffe train -solver examples/mnist/lenet_solver.prototxt
(2)-gpu 指定哪一塊gpu來執行,如果是全部則是 -gpu all,例如
./build/tools/caffe train -solver examples/mnist/lenet_solver.prototxt -gpu 2
(3)-snapshot:可選引數。該引數用來從快照(snapshot)中恢復訓練,例如
./build/tools/caffe train -solver examples/mnist/lenet_solver.prototxt -snapshot examples/mnist/lenet_iter_5000.solverstate
(4)-weights:可選引數。用預先訓練好的權重來fine-tuning模型,需要乙個caffemodel,不能和-snapshot同時使用。如:
./build/tools/caffe train -solver examples/finetuning_on_flickr_style/solver.prototxt -weights models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel
(5)-iterations: 可選引數,迭代次數,預設為50
(6)-model: 可選引數,定義為protocal buffer中的模型
(7)-sighup_effect:可選引數。用來設定當程式發生
掛起事件時,執行的操作,可以設定為snapshot, stop或none, 預設為snapshot
(8)-sigint_effect: 可選引數。用來設定當程式發生鍵盤中止事件時(ctrl+c), 執行的操作,可以設定為snapshot, stop或none, 預設為stop
可以在模型配置檔案中定義輸入為accuracy還是loss,如果要驗證已經訓練好的模型,則可以用如下命令:
./build/tools/caffe test -model examples/mnist/lenet_train_test.prototxt -weights examples/mnist/lenet_iter_10000.caffemodel -gpu 0 -iterations 100
./build/tools/caffe time -model examples/mnist/lenet_train_test.prototxt -iterations 10
顯示模型迭代10次所用的時間,包括forward和backward所用的時間,也包括每層forwand和backward所用的平均時間。
./build/tools/caffe device_query -gpu 0
最後,在linux下,本身就有乙個time命令,因此可以結合進來使用,因此我們執行mnist例子的最終命令是(一塊gpu):
sudo time ./build/toos/caffe train -solver examples/mnist/lenet_solver.prototxt
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 ...