caffe命令及其引數解析

2021-10-02 20:39:42 字數 1959 閱讀 2453

caffe程式的命令列執行格式如下:

caffe

包含如下四種:

train————(訓練或finetune模型(model))

test————(測試模型)

device_query————(顯示gpu資訊)

time————(顯示程式執行時間)

引數包含:

-solver

-gpu

-snapshot

-weights

-iteration

-model

-sighup_effect

-sigint_effect

-solver:必選引數。乙個protocol buffer型別的檔案,即模型的配置檔案。如:

./build/tools/caffe train -solver examples/mnist/lenet_solver.prototxt

-gpu: 可選引數。該引數用來指定用哪一塊gpu執行,根據gpu的id進行選擇,如果設定為'-gpu all'則使用所有的gpu執行。如使用第二塊gpu執行:

./build/tools/caffe train -solver examples/mnist/lenet_solver.prototxt -gpu 1

-snapshot:可選引數。該引數用來從快照(snapshot)中恢復訓練。如:

./build/tools/caffe train -solver examples/mnist/lenet_solver.prototxt -snapshot examples/mnist/lenet_iter_5000.solverstate

-weights:可選引數。用預先訓練好的權重來fine-tuning模型,需要乙個caffemodel,不能和-snapshot同時使用。如:

./build/tools/caffe train -solver examples/solver.prototxt -weights    models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel

-iterations: 可選引數,迭代次數,預設為50。 

-model:可選引數,定義在protocol buffer檔案中的模型。也可以在solver配置檔案中指定。

-sighup_effect:可選引數。用來設定當程式發生掛起事件時,執行的操作,可以設定為snapshot, stop或none, 預設為snapshot

-sigint_effect: 可選引數。用來設定當程式發生鍵盤中止事件時(ctrl+c), 執行的操作,可以設定為snapshot, stop或none, 預設為stop

test引數用在測試階段:

./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 100

這個例子用來在螢幕上顯示lenet模型迭代10次所使用的時間。包括每次迭代的forward和backward所用的時間,也包括每層forward和backward所用的平均時間。

./build/tools/caffe time -model examples/mnist/lenet_train_test.prototxt -weights examples/mnist/lenet_iter_10000.caffemodel -gpu 0 -iterations 10

利用給定的權重,利用第一塊gpu,迭代10次lenet模型所用的時間。

device_query引數用來診斷gpu資訊。

./build/tools/caffe device_query -gpu 0

caffe命令及其引數解析

caffe的c 主程式 caffe.cpp 放在根目錄下的tools資料夾內,當然還有一些其它的功能檔案,如 convert imageset.cpp,train net.cpp,test net.cpp等也放在這個資料夾內。經過編譯後,這些檔案都被編譯成了可執行檔案,放在了 build tools...

caffe 命令及引數解析

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

caffe命令列解析

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