本站的內容是訓練自己的網絡結構來分類鳥和狗。

1.準備自己的數據集。

? ? ? ?1.1 ?在caffe的data文件夾下面建立一個myself文件夾。在myself文件夾下面建立train和test文件夾。

?

? ? ? ?1.2 百度下載10張鳥的圖片和10張狗的圖片最為訓練集。再下載4張鳥的圖片和4張狗的圖片最為驗證集。（只是為了給大家演示一下就沒有弄太多的數據集）。對訓練集圖片進行順序編號dog1.jpeg，dog2.jpeg .... dog10.jpeg。bird1.jpeg ?bird2.jpeg ... ?bird10.jpeg。對驗證集的圖片進行編號?dog1.jpeg，dog2.jpeg .... dog4.jpeg。bird1.jpeg ?bird2.jpeg ... ?bird4.jpeg。 將訓練集圖片復制到train文件夾下面。驗證集圖片復制到test文件夾下面。

? ? ? ?1.3 在myself文件夾下面新建train.txt 和 test.txt文本文檔。0表示bird類別 ?1表示dog類別。

? ? ? ? ? ??

? ? ? ? ? ??

2.制作數據集和均值文件：

? ? 2.1 在caffe的examples文件夾下面建立一個 myself 文件。找到caffe -> examples -> imagenet文件夾，將里面的create_imagenet.sh 、make_imagenet_mean.sh 、train_caffenet.sh 這三個文件復制到 examples文件夾下面的myself文件夾里面。

? ? ?2.2 修改create_imagenet.sh 將里面的路徑設置文自己的路徑。下面是我修改的路徑。（只修改了紅色圈住的地方，將前面的路徑修改成自己的絕對路徑，然后RESIZE=TRUE，是為了將圖片都resize成統一大小的圖片。）設置好之后運行sh文件就可以在examples->myself文件夾下面得到生成的兩個lmdb文件夾。

? ? ?

? ? 2.3生成均值文件：修改make_imagenet_mean.sh ? 只是修改路徑 把有的路徑修改成自己的絕對路徑。修改完之后運行sh文件，就可以在myself文件夾下面得到一個 ?imagenet_mean.binaryproto 文件。

? ? ??

3.訓練自己的網絡：（為了方便起見我直接拿caffenet的網絡結構來訓練，你可以把你的網絡結構復制進來替換掉caffenet網絡）找到 caffe-> models -> bvlc_reference_caffenet ?將里面的solver.prototxt 和 train_val.prototxt兩個文件復制到 examples->mtself 文件夾下面。 ?修改train_val.prototxt文件（紅框修改成自己的文件夾路徑，綠框為batch_size大小由于數據集太少建議設置成1.）

修改solver.prototxt文件：（換成自己的路徑）

修改train_caffenet.sh文件（修改成自己的路徑）：

修改完之后直接 ?bash ?train_caffenet.sh 就可以訓練網絡了。

ps：你可以修改train_val.prototxt將里面的網絡結構修改成你自己的網絡來訓練。

訓練結果：

總結

以上是生活随笔為你收集整理的Caffe教程：训练自己的网络结构来分类。的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。