MXNET数据集制作,生成rec文件

https://github.com/apache/incubator-mxnet/blob/master/tools/im2rec.py
(一)制作rec数据集
(1)下载im2rec.py 文件
(2)python im2rec.py train 文件地址(内部按类别分成不同的文件夹) --list --recursive
生成 .lst文件。
(3)python im2rec.py train E:\pcb_defect_image\2018_11_15\all_40_40\train --num-thread=4
生成 .rec文件和 .idx文件。
(二)读取rec数据集
(1)mx.io.ImageRecordIter

你可能感兴趣的:(python,mxnet)