1、生成.lst文件
制作一个文件路径和标签的列表:
import os import sys #第一个参数是输入路径 input_path=sys.argv[1].rstrip(os.sep) #第二个参数是输出路径 output_path=sys.argv[2] #列出输入文件夹下所有文件名 filenames=os.listdir(input_path) with open(output_path,'w')as f: for i,filename in enumerate(filenames): filepath=os.sep.join([input_path,filename]) label=filename[:filename.rfind('.')].split('.')[1] line='{} {} {} '.format(i,label,filepath) f.write(line)
将代码保存为mxnet_imglist.py,然后执行下面命令:
python mxnet_imglist.py mnist/train train.lst
python mxnet_imglist.py mnist/val val.lst
python mxnet_imglist.py mnist/test test.lst
2、生成.rec文件
利用MXNet的官方工具mxnet oolsim2rec.py:
文件格式如下:
执行命令:python mxnet oolsim2rec.py lst mnist