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_