1. 程式人生 > >【深度學習框架Caffe學習與應用】第三課 將圖片資料轉化為LMDB資料``

【深度學習框架Caffe學習與應用】第三課 將圖片資料轉化為LMDB資料``

1.將圖片資料轉化為LMDB資料

第一步:建立圖片檔案列表清單,一般為一個txt檔案,一行一張圖片

我在caffe/data/目錄下新建一個test_data的資料夾,裡面放訓練集及資料集



首先看一下data/test_data/train/male中的圖片,顯示出檔名


然後終端執行下面命令,將資料夾中圖片的名稱寫到txt檔案中


內容如下圖所示


同樣的,生成訓練集的標籤檔案

這裡說一下,生成標籤命令的解釋:

ls train/female/表示female資料夾下的圖片名稱,sed "s:^:male/:"表示txt檔案中,開頭是male/,sed "s:$:   0:"表示圖片名之後,加上   0(有空格),>> t_train.txt,表示輸出到這個檔案中。

第二步:使用Caffe工具命令convert_imageset

該工具通過命令列方式使用,命令列的格式如下:
convert_imageset [FLAGS] [ROOTFOLDER/] [LISTFILE] [DB_NAME ]


我這裡一直顯示,沒有convert_imageset命令,這是因為當前目錄為:~/caffe/data/test_data,而convert_imageset所在目錄為:~caffe/build/tools/,所以會找不到該命令,我把convert_imageset放到當前檔案件test_data下,執行命令如下,成功:


或者回到caffe根目錄下,執行以下程式碼也可以: