1. 程式人生 > >Windows+caffe對自己的樣本進行特徵提取

Windows+caffe對自己的樣本進行特徵提取

生成lmdb/leveldb檔案+均值檔案

利用caffe對自己的圖片生成leveldb格式檔案:
http://www.cnblogs.com/TensorSense/p/6744075.html
1、將圖片進行分類,建立train和test資料夾存放訓練和測試的影象
2、建立txt標籤檔案存放圖片路徑以及類別
這裡寫圖片描述
其中,需要注意的是,——在檔名定義的時候標籤與檔名之間只能有一個空格,多了會產生錯誤。
——在使用路徑時,注意用絕對路徑還是相對路徑
3、新建.bat檔案呼叫convert_imageset.exe轉化圖片格式:
D:\caffe\caffe-master\Build\x64\Release\convert_imageset.exe --resize_width=64 --resize_height=64 ./sub1/ train.txt train_leveldb -backend=leveldb
D:\caffe\caffe-master\Build\x64\Release\convert_imageset.exe --resize_width=28 --resize_height=28 ./sub2/ test.txt test_leveldb -backend=leveldb
Pause

這裡的每行包含七個部分:
第一個部分:D:\caffe-master\Build\x64\Release\convert_imageset.exe 表示 convert_imageset.exe所在路徑(PS,若是用Debug生成的,則在將Release改成Debug即可)
第二、三部分,是對原始圖片進行了resize
第四部分:
./train/ 是訓練圖片檔案所在路徑,此處用了相對路徑的形式(不瞭解相對路徑的同學,請度娘 ./ ../的作用)
第五部分,train_labels.txt是訓練集的標籤
第六部分:train_lmdb 是生成LMDB格式檔案的檔名
第七部分:-backend=lmdb 是所要轉成的格式(要轉成 leveldb的則改成 -backend=leveldb)
然後雙擊執行

執行正常的話,在資料夾下會出現 test_leveldb 和train_leveldb 兩個資料夾。至此,已經將自己的資料集轉換成為leveldb格式的資料了,即caffe可讀的資料格式。(將檔案中leveldb換為lmdb即可更改格式)
雙擊.bat檔案執行之後生成檔案:
這裡寫圖片描述

對生成的train_lmdb進行均值檔案獲取:
SET GLOG_logtostderr=1
compute_image_mean.exe train_lmdb
image_mean.binaryproto
pause
我是直接在cmd中執行的,生成了image_mean.binaryproto 檔案

對生成檔案進行特徵提取