1. 程式人生 > >用caffe將自己的影象資料轉換成lmdb

用caffe將自己的影象資料轉換成lmdb

前提:

使用convert_imageset.exe的前提是成功編譯caffe,在編譯成功caffe後,可將編譯caffe時的caffe.cpp去掉,設定生成方式為lib,生成lib檔案,供其他的函式使用,如下圖:


生成caffe.lib後,在編譯convert_imageset.cpp時可以在附加依賴項中加上caffe.lib,在C++前處理器中加上USE_LMDB和USE_OPENCV,如圖


成功生成convert_imageset.exe後,準備將自己的影象資料變為lmdb格式。

1、準備自己的影象資料,放到一個資料夾下,例如NSP_Logo_HOG,我的資料夾,然後生成一個描述檔案

pos_hog.txt,內容是影象檔名和影象類別,然後準備一個儲存lmdb資料的資料夾traindb,如圖:


然後寫一個批處理檔案convert.bat:(批處理檔案中的描述檔案應為pos_hog.txt)

內容為:


上面的內容根據自己的調整即可,執行bat檔案,結果為:


下一步就是去掉影象的均值,在另一篇檔案中介紹。

注:我的影象資料都已經進行了歸一化,否則的話在轉換中加上歸一化引數寬和高,可在bat檔案中使用

Convert.exe use -help來檢視Convert.exe的使用方法。