从mxnet的rec文件中恢复jpg格式

网上清一色“互相参考”的回答……：

 这里迷之501195，然后搜了几个答案居然都写的501195.......

这里给出另一份答案：

参考：

cvtower/seesawfacenet_pytorch: offical seesawfacenet pytorch implement, https://arxiv.org/abs/1908.09124 (github.com)

def load_mx_rec(rec_path):
    # save_path: 转为图片后存储的位置
    # rec_path: rec文件的路径
    save_path = './ms1m_ibug'
    save_path = os.path.join(save_path, 'imgs')
    if not os.path.exists(save_path):
        os.makedirs(save_path)
    imgrec = mx.recordio.MXIndexedRecordIO(os.path.join(rec_path,'train.idx'), os.path.join(rec_path,'train.rec'), 'r')
    img_info = imgrec.read_idx(0)
    header,_ = mx.recordio.unpack(img_info)
    max_idx = int(header.label[0])
    print('max_idx:', max_idx) # 这里的max_idx指的是图片的个数
    for idx in tqdm(range(1,max_idx)):
        img_info = imgrec.read_idx(idx)
        header, img = mx.recordio.unpack(img_info)
        label = int(header.label)
        img = mx.image.imdecode(img).asnumpy() # mxnet格式文件转为numpy
        img = Image.fromarray(img) # numpy转为Image方便保存
        label_path = os.path.join(save_path,str(label))
        if not os.path.exists(label_path):
            os.makedirs(label_path)
        img.save(os.path.join(label_path,'{}.jpg'.format(idx)), quality=95)