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表示將圖片縮放到固定大小,這裡為 .
–num-thread表示採用多個執行緒處理.
處理後得到檔案,
eye_train.idx
eye_train.rec
eye_val.idx
eye_val.rec