第九节,MXNet:用im2rec.py将图像打包生成.rec文件
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='{}\t{}\t{}\n'.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\tools\im2rec.py:
文件格式如下:
执行命令:python mxnet\tools\im2rec.py lst mnist
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步