mxnet-im2rec.py文件训练数据生成方法

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/u011961856/article/details/79209652

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

猜你喜欢

转载自blog.csdn.net/u011961856/article/details/79209652