1)調用solver.prototxt生成.caffemodel和.solverstate
caffe中的bin檔案夾裡有caffe.exe,在上級目錄中建立.bat檔案,添加内容并儲存運作:
./bin/caffe.exe train --solver=examples/cifar10/cifar10_quick_solver.prototxtpause
然後就會在examples/cifar10/檔案夾中,訓練出.caffemodel和.solverstate檔案,其中.caffemodel用于測試,.solverstate用于恢複訓練(像斷點續 傳)
2)當所有資料都訓練好之後,接下來就是如何将模型應用到實際資料了:
./build/tools/caffe.bin test -model=examples/mnist/lenet_train_test.prototxt -weights=examples/mnist/lenet_iter_10000.caffemodel -gpu=0
如果沒有GPU則使用
./build/tools/caffe.bin test -model=examples/mnist/lenet_train_test.prototxt -weights=examples/mnist/lenet_iter_10000.caffemodel
test:表示對訓練好的模型進行Testing,而不是training。其他參數包括train, time, device_query。
-model=XXX:指定模型prototxt檔案,這是一個文本檔案,較長的描述了網絡結構和資料集資訊