第九节，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

第九节，MXNet：用im2rec.py将图像打包生成.rec文件

推荐阅读