1. 程式人生 > >MXNET資料集製作,生成rec檔案

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