1. 程式人生 > >mxnet-im2rec.py檔案訓練資料生成方法

mxnet-im2rec.py檔案訓練資料生成方法

生成list檔案

python /home/research/tools/incubator-mxnet/tools/im2rec.py –list 1 –recursive 1 –train-ratio 0.9 eye eye/

輸入引數含義:

–list:當要生成list檔案時,這個引數一定要設為True,表示當前用來生成的list檔案;預設是生成rec檔案;

–recursive:遞迴的遍歷你的所有資料集,要設為True;

–train-ratio:用來將你的全部資料集拆分成兩部分:訓練集(train)和交叉驗證集(val),具體多少作為訓練集,多少作為驗證集,就由這個引數來確定;

–test-ratio:同上,分成訓練集和測試集兩部分;

prefix:這裡指的是你要生成list檔案的字首名,上述命令列為 eye;

root:這裡指的是圖片資料存放路徑,上述命令列為 eye/;

–exts:讀取圖片字尾,關於圖片格式,檢視程式碼im2rec.py可以看到,

rgroup.add_argument(‘–encoding’, type=str, default=’.jpg’, choices=[‘.jpg’, ‘.png’],

也就是當前版本支援圖片格式為jpg,png.

–label_width://標籤個數,預設為1,即每個圖片只有一個標籤

處理後得到檔案:

eye_val.lst,eye_train.lst

.lst檔案每行儲存了:編號,label,圖片讀取路徑,例如下面為生成eye_train.lst的某幾行的內容,

178 0.000000 close/leftclosed_eye_1340.BMP_face_1.jpg
3255 1.000000 open/rightcf959e57061c5fa325261c3e5e219bbe.jpg
1376 0.000000 close/rightu=2728579787,3784543441&fm=200&gp=0.jpg
2773 1.000000 open/right0e79e7fdac30c9643458fe0926ecca2c.jpg
193 0.000000 close/leftclosed_eye_1509.jpg_face_6.jpg
2802 1.000000 open/right19fdf6fa9463ec6c359332364c64c6bb.jpg
3330 1.000000 open/rightefb70004cbc13c27f4f46d87e158a770.jpg
1066 0.000000 close/rightclosed_eye_2440.jpg_face_1.jpg
1096 0.000000 close/rightclosed_eye_2767.BMP_face_1.jpg

生成rec檔案

python /home/research/tools/incubator-mxnet/tools/im2rec.py –resize 39 –num-thread 4 eye eye/

–resize表示將圖片縮放到固定大小,這裡為 39 × 39 .

–num-thread表示採用多個執行緒處理.

處理後得到檔案,

eye_train.idx

eye_train.rec

eye_val.idx

eye_val.rec