生成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