caffe如何進行數據集測試??
caffe中.prototxt文件里設置了每迭代多少次測試一次,實際上這個最終模型測試的結果是驗證集的結果,如何對數據集進行測試呢???用自帶的caffe -test嗎?還是用matlab介面去一張張的測???
謝邀。直接使用caffe test就可以,腳本的寫法是這樣的:
#!/usr/bin/env sh
GLOG_logtostderr=1 ./build/tools/caffe.bin test --model=/path/to/your/prototxt --weights=/path/to/your/caffemodel
我使用的是caffe-windows,使用這個命令即可:caffe test -model=test_prototxt -weights=model.caffemodel -iterations=n -gpu 0
其中model是配置文件,可以指定測試集路徑,weights就是訓練出的模型,iterations是迭代次數,默認為50次.
舉個栗子:mnist數據集訓練完畢後,得到了lenet_iter_10000.caffemodel,如果想要在測試集上測試,只需要將lenet_train_test.prototxt中驗證集路徑改成你的測試集路徑即可:ok,接下來進行測試:caffe test -model=lenet_train_test.prototxt weights=lenet_iter_10000.caffemodel -iterations=100 -gpu 0
由於測試batch_size為100,所以迭代次數設置為100,這樣就可以測試完所有的10000張圖片,結果如下:
已解決
用自帶的caffe/build/tools/caffe test -model xxx.prototxt -weights xxx.caffemodel -iterations n -gpu x,由於我用的是cpu進行運行caffe,所以沒有-gpu選項,由於xxx.prototxt作為訓練和測試模型使用,裡面已經包括訓練集和測試集的數據,因此就直接使用caffe命令進行運行了。
推薦閱讀:
※caffe 每個樣本對應多個label?
※caffe的finetuning是如何更新網路參數的?
※CNN全連接層隱層結點數(output_dim)的設定有什麼講究嗎?
※Ubuntu這麼難安裝嗎?
TAG:深度學習DeepLearning | Caffe深度學習框架 |