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