本站的內(nèi)容是訓(xùn)練自己的網(wǎng)絡(luò)結(jié)構(gòu)來(lái)分類鳥和狗。

1.準(zhǔn)備自己的數(shù)據(jù)集。

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

?

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

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

? ? ? ? ? ??

? ? ? ? ? ??

2.制作數(shù)據(jù)集和均值文件：

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

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

? ? ?

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

? ? ??

3.訓(xùn)練自己的網(wǎng)絡(luò)：（為了方便起見(jiàn)我直接拿caffenet的網(wǎng)絡(luò)結(jié)構(gòu)來(lái)訓(xùn)練，你可以把你的網(wǎng)絡(luò)結(jié)構(gòu)復(fù)制進(jìn)來(lái)替換掉caffenet網(wǎng)絡(luò)）找到 caffe-> models -> bvlc_reference_caffenet ?將里面的solver.prototxt 和 train_val.prototxt兩個(gè)文件復(fù)制到 examples->mtself 文件夾下面。 ?修改train_val.prototxt文件（紅框修改成自己的文件夾路徑，綠框?yàn)閎atch_size大小由于數(shù)據(jù)集太少建議設(shè)置成1.）

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

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

修改完之后直接 ?bash ?train_caffenet.sh 就可以訓(xùn)練網(wǎng)絡(luò)了。

ps：你可以修改train_val.prototxt將里面的網(wǎng)絡(luò)結(jié)構(gòu)修改成你自己的網(wǎng)絡(luò)來(lái)訓(xùn)練。

訓(xùn)練結(jié)果：

總結(jié)

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

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。