Visual Studio2013 caffe MNIST測試詳細步驟
想知道自己的caffe是否安裝成功,或者想體驗一下caffe,可以繼續往下看:
使用MNIST數據集的概要:
首先你要有生成的exe文件,然後下載MNIST數據集,設置相應的參數,利用數據集訓練網絡,用生成的網絡測試樣本。
1.生成exe文件
步驟:解決方案(右鍵)--屬性—多啓動項目—配置設爲releaseX64—“啓動”
1.1設置多個啓動項目前需要按照網上的教程先生成(build)libcaffe
1.2設置多個啓動項目,同時生成多.EXE文件
1.3修改啓動 release X64
這樣生成的exe纔會放到release文件夾下,否則爲debug文件夾。區別可以百度debug和release的區別
1.4生成的結果如下圖所示,有多個exe文件:
2.測試MNIST數據
步驟:下載MNIST數據集—設置lenet_solver.prototxt和lenet_train_test.prototxt文件參數—編輯訓練文件(train_lenet.bat…) 編輯測試文件(….)--運行訓練文件—運行測試文件
2.1下載MNIST數據集
將下載的MNIST數據集放到\example\mnist文件夾下,如圖,網上有很多的下載渠道,百度一下就出來了,下載下來的數據多爲leveldb格式,如下圖中的後綴名爲leveldb,有一個train和test數據集(使用lmdb格式要自己轉一下,使用leveldb可以繼續看下去):
2.2修改lenet_solver文件
這裏主要是設置:1>定義參數文件位置(數據集名稱、位置不能錯) 2>運行模式選擇GPU還是CPU,小數據集,CPU和GPU速度相差不大,當在大數據集上運行時,這種差距就會很顯著,GPU更快。
2.3修改lenet_train_test.prototxt文件
這裏主要設置source和backend。Source爲MNIST數據集的位置(有一個train和test數據集);backend爲數據格式,因爲我們下載的是leveldb數據集,所以要將其改爲leveldb(數據集名稱、位置不能錯)。
2.4編輯訓練文件
可以用lenet_train.sh,或者自己編寫一個bat文件,就是很多教程中說的train_lenet.bat(也行有些caffe包自帶),我下載的代碼中沒有這個文件,但是可以自己寫,也可以隨意命名,爲.bat文件即可(可用記事本,完成後改.bat後綴名)。這裏的位置用的絕對路徑,也可以用相對路徑。
這兩個位置不能搞錯,中間的分隔符爲“/”。
測試過程
訓練成功後,在MNIST文件夾下會出現這幾個文件:
2.5編輯測試文件
規則與訓練文件相似,我還是自己寫的一個bat文件,將caffe.exe,lenet_train_test.prototxt,lenet_iter_10000.caffemodel三個文件及位置放入,紅色框內抄下來,如下所示:
運行結果:
在測試過程覺得寫得很詳細的博客:
caffe安裝:http://www.cnblogs.com/love6tao/p/5706830.html
MNIST測試:https://www.cnblogs.com/codingmengmeng/p/6184393.html
https://www.cnblogs.com/yixuan-xu/p/5858595.html
ps:
文中的路徑文件名一定要正確,出錯的時候看DOS框提示找不到文件,一定要仔細檢查!
MNIST中有編輯好的網絡,可以不用自己寫,如果只是想測試一下,可以按照默認的來,簡單改幾個位置即可。