天天看點

.caffemodel的生成和利用

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檔案,這是一個文本檔案,較長的描述了網絡結構和資料集資訊

繼續閱讀