ubuntu14.04 caffe 製作自己的資料
阿新 • • 發佈:2019-01-05
嘿嘿,最近部落格更得有點勤啊,主要博主最近快畢業了,閒來無事就沒事多學點東西把。。。
博主昨天把最難的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目錄中檢視生成的檔案了。
注意的是,地址什麼一定要相對應,不然會報各種奇怪的錯誤。