在mxnet进行数据输入,需先转化成mxnet对应的数据格式–rec格式。
- 生成lst文件
进入mxnet/tools文件夹中,运行im2rec.py 文件生成lst文件
python im2rec.py --list --recursive datatrain /home/datasets/
其中:
–list True :这表示生成lst文件
–recursive True :表示读取该目录下包括子目录下的所以文件
datatrain :即prefix, 生成的lst文件前缀,可自拟
/home/datasets/ :对应路径, 即包含数据集image的路径
注意:im2rec.py默认的图片格式为jpg和jpeg,如果是其他格式,需要修改
执行完后,会生成datatrain.lst文件
- 生成rec文件
Python im2rec.py datatrain /home/datasets/
datatrain :即prefix, 生成的rec文件前缀,可自拟
/home/datasets/ :对应路径, 即包含数据集image的路径
执行完后,会生成datatrain.rec文件