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

 

 

posted @   hengxin  阅读(1272)  评论(0编辑  收藏  举报
努力加载评论中...
点击右上角即可分享
微信分享提示