python 读取mxnet recordio

Posted 2021-05-09

from mxnet import recordio
import mxnet as mx
import matplotlib.pyplot as plt
path_imgidx = 'D:/code/DeepLearning/Face/InsightFace/insightface/datasets/faces_ms1m_112x112/train.idx'
path_imgrec = 'D:/code/DeepLearning/Face/InsightFace/insightface/datasets/faces_ms1m_112x112/train.rec'

imgrec = recordio.MXIndexedRecordIO(path_imgidx, path_imgrec, 'r')
#%% 1 ~ 3804847
for i in range(3804846):
    header, s = recordio.unpack(imgrec.read_idx(i+1))
    img = mx.image.imdecode(s).asnumpy() 
    plt.imshow(img)
    plt.title('id=' + str(i) + 'label=' + str(header.label))
    plt.pause(0.1)