可以下載我的練習數據:http://pan.baidu.com/s/1MotUe,放在data/re根目錄下。
這些數據共有500張圖片,400張用於訓練,100張用於測試。分爲大巴車、恐龍、大象、鮮花和馬五個類,每個類100張。
在examples/根目錄下新建文件夾 mydata1,在此文件夾內放入訓練集、測試集等相關文件。
在examples/mydata1根目錄下,新建imgtrainset文件夾 和 train.txt 文本文件
(注意: txt文件內,類別必須從0開始標註,不能從3開始)
新建convert_image_traindata.bat,內容如下:
SET GLOG_logtostderr=1
E:\study_materials\Caffe\caffe-master\caffe-master\Build\x64\Release\convert_imageset.exe imgtrainset/ train.txt train_modelre_lmdb -resize_width=227 -resize_height=227--backend=lmdb --gray=false--shuffle=true
pause
則在examples/mydata1根目錄下,就會多一個文件:train_modelre_lmdb
在examples/mydata1根目錄下,新建imgtestset文件夾 和 val.txt 文本文件
(同樣的,txt文件內必須從0開始)
新建convert_image_testdata.bat,內容如下:
SET GLOG_logtostderr=1
E:\study_materials\Caffe\caffe-master\caffe-master\Build\x64\Release\convert_imageset.exe imgtestset/ val.txt test_modelre_lmdb -resize_width=227 -resize_height=227 --backend=lmdb --gray=false --shuffle=true
pause
則在examples/mydata1根目錄下,就會多一個文件:test_modelre_lmdb
在examples/mydata1根目錄下新建image_mean_train.bat ,內容如下:
E:\study_materials\Caffe\caffe-master\caffe-master\Build\x64\Release\compute_image_mean.exe E:\study_materials\Caffe\caffe-master\caffe-master\examples\mydata1\train_modelre_lmdb E:\study_materials\Caffe\caffe-master\caffe-master\examples\mydata1\modelre_train_mean.binaryproto --backend=lmdb
則在examples/mydata1根目錄下,就會多一個文件:modelre_train_mean.binaryproto
參考資料:
http://blog.csdn.net/helloworldding/article/details/76559855
https://www.cnblogs.com/louyihang-loves-baiyan/p/5038758.html
學習率變化規律我們設置爲隨着迭代次數的增加,慢慢變低。總共迭代1000次,我們將變化5次,所以stepsize設置爲1000/5=200,即每迭代200次,我們就降低一次學習率。
(注:stepsize不能太小,如果太小會導致學習率再後來越來越小,達不到充分收斂的效果)