本教程盡量詳細,大多步驟都有圖,如果運行出錯,請先對照自己的文件是否和圖上的一樣,包括標點啊,空格啊,斜杠,反斜杠啊之類的小細節。
本例程是在 win10 64位 caffe-master vs2013下進行的,並且已經配置GPU版本,若用CPU,則在cifar10_quick_solver.prototxt中更改
# solver mode: CPU or GPU
solver_mode: CPU
摘要:win10下的caffe例子:cifar10
第一步:
官網下載bin格式的cifar10 : http://www.cs.toronto.edu/~kriz/cifar.html
解壓存放在 :caffe-master\examples\cifar10\input_folder當中(input_folder文件夾需要自己創建)
如圖:
第二步:
在caffe-master\examples\cifar10 下創建一個記事本(后綴改為 .bat ),輸入以下code
..\..\Build\x64\Release\convert_cifar_data.exe convert_cifar_data.exe input_folder output_folders leveldb
pause
保存為convert_cifar10.bat (后綴改為 .bat)
如圖:
然后雙擊運行,可以看到在 caffe-master\examples\cifar10下會生成一個文件夾,output_folders。里面存放的就是轉換好的數據
第三步:
計算出均值文件:mean.binaryproto
首先要確保在caffe-master\Build\x64\Release 下有compute_image_mean.exe ,沒有的話請打開Caffe.sln,將compute_image_mean 設置為 啟動項目,然后ctrl+F5,等待編譯結束,在caffe-master\Build\x64\Release下就有compute_image_mean.exe了
如圖:
有了compute_image_mean.exe,我們要在caffe-master\examples\cifar10下創建一個記事本(后綴改為 .bat ),輸入以下code
..\..\Build\x64\Release\compute_image_mean.exe-backend=leveldb ../../examples\cifar10\output_folders\cifar10_train_leveldbmean.binaryproto
Pause
然后運行此文件,在caffe-master\examples\cifar10 下會出現mean.binaryproto文件
第四步:
更改caffe-master\examples\cifar10 下的cifar10_quick_solver.prototxt和 cifar10_quick_train_test.prototxt 主要是改路徑
我的cifar10_quick_solver.prototxt改中了:
net:"../../examples/cifar10/cifar10_quick_train_test.prototxt"
snapshot_prefix:"../../examples/cifar10/cifar10_quick"
cifar10_quick_train_test.prototxt中改了:
mean_file:"../../examples/cifar10/mean.binaryproto"
source:"../../examples/cifar10/output_folders/cifar10_train_leveldb"
mean_file:"../../examples/cifar10/mean.binaryproto"
source:"../../examples/cifar10/output_folders/cifar10_test_leveldb"
以及 backend: LEVELDB (有兩處喔!)
第五步:
更改好兩個文件之后就可以運行caffe來訓練啦,提供兩種方法,一種是編寫批處理文件bat,一種是直接在vs2013中運行
先說第一種,在caffe-master\examples\cifar10下創建文本文件,更改后綴為.bat,復制以下代碼:
..\..\Build\x64\Release\caffe.exe train--solver=../../examples/cifar10/cifar10_quick_solver.prototxt
Pause
如圖:
保存之后,運行即可。
第二種方法是在vs2013里進行的,先將caffe設置為啟動項目:
如圖:
再點擊屬性,找到調試,在命令參數中輸入:train --solver=../../examples/cifar10/cifar10_quick_solver.prototxt
如圖:
點擊確定,ctrl+F5 即可