caffe怎麼test訓練好的model?
01-23
輸入一張圖片,然後給出預測,這就是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深度學習框架 |