1. 程式人生 > >ubuntu14.04 caffe 製作自己的資料

ubuntu14.04 caffe 製作自己的資料

嘿嘿,最近部落格更得有點勤啊,主要博主最近快畢業了,閒來無事就沒事多學點東西把。。。

博主昨天把最難的nvidia驅動這坑給跳過了,但是cuda下載老是失敗。。。。。。fuck the nvidia!!!

今天博主想自己做caffe的資料集,來給出詳細的步驟。

首先切換到我的caffe的主目錄,然後

cd data
我在data目錄下建立了一個叫myself的資料夾

我建立瞭如下的檔案

我來說明一下,test檔案放我們需要的圖片,測試用。train資料夾存放我們訓練用的檔案。test和train的txt檔案分別放對應資料夾的名字和圖片的標籤,我這裡為了簡單,只放了幾張貓狗的圖片,並且標籤用0-1來表示。具體表示如下:

我們可以看到,這裡的第一行為圖片的id,然後加個空格,然後放對應圖片的標籤。最後儲存。

需要注意的是,儲存完請確認資料的行數,我之前不知怎麼就是多一行,可能儲存的時候游標換行了,所以一直提示找不到檔案。。。

然後我們把caffe裡面自帶的creatimagenet的驅動檔案開啟

修改如下地方

修改好之後,我們進入caffe的主目錄

然後

./data/myself/create_imagenet.sh

這樣就可以在example/myself目錄中檢視生成的檔案了。

注意的是,地址什麼一定要相對應,不然會報各種奇怪的錯誤。