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深度學習框架 |