caffe怎麼test訓練好的model?

輸入一張圖片,然後給出預測,這就是test嗎?

以下這段代碼又是什麼意思?


./build/tools/caffe.bin test

-model=examples/mnist/lenet_train_test.prototxt

-weights=examples/mnist/lenet_iter_10000.caffemodel

-gpu=0


感謝題主這行代碼!知道怎麼使用caffemodel進行測試了,已經把詳細步驟截圖記錄。我把caffe跑mnist的步驟寫成了技術博客發在我的csdn博客上,不過還在審核,所以一時半會兒看不到,在這裡先回答題主的問題《如何使用訓練後的網路lenet_iter_5000.caffemodel與lenet_iter_10000.caffemodel?》

1、在網路訓練完畢後,將在指定路徑後生成四個文件,其中兩個caffemodel文件lenet_iter_5000.caffemodel與lenet_iter_10000.caffemodel為我們測試時所需要的,它們分別存放著程序迭代5000次與10000次後網路的參數。

為了防止連自己都不知道這兩個文件將在哪裡生成,我們首先需要在訓練之前就先設置好caffemodel文件的生成路徑,如圖:

然後接下來等待網路訓練完畢後,將在對應的路徑下生成四個文件:

2、在CAFFE/examples/mnist目錄下新建一個文本文件,然後將後綴名改成sh。填入以下語句:

注意,這樣填寫只是為了便於觀看,但直接保存後是不能跑起來的,必須把第2~5條語句全部放在同一行,語句與語句之間用一個空格隔開。

3、保存該sh文件,然後雙擊,即可出現訓練情況:

2016.5.7

by 悠望南山

(註:如要轉載,請保留知乎或csdn博客地址CAFFE學習筆記(二)Caffe_Example之測試mnist,謝謝!)


這是用命令行進行測試,也可以寫python、matlab或C++程序進行測試。

  • ./build/tools/caffe.bin test 表示只做預測(前向傳播計算),不進行參數更新(後向傳播計算)

  • -model=examples/mnist/lenet_train_test.prototxt 指定模型描述文本文件

  • -weights=examples/mnist/lenet_iter_10000.caffemodel 指定訓練好的模型文件

  • -gpu=0 指定運行設備

除了 -gpu=0 外,內容參考自《深度學習:21天實戰caffe》第76頁


同問,回歸的怎麼寫?


推薦閱讀:

caffe 每個樣本對應多個label?
caffe的finetuning是如何更新網路參數的?
caffe如何進行數據集測試??
CNN全連接層隱層結點數(output_dim)的設定有什麼講究嗎?
Ubuntu這麼難安裝嗎?

TAG:機器學習 | 神經網路 | 深度學習DeepLearning | Caffe深度學習框架 |