使用mxnet还原jpg格式的人脸识别数据集

import mxnet as mx
import argparse
import PIL.Image
import os
from mxnet import ndarray as nd


def parse_args():
    parser = argparse.ArgumentParser(
        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
        description='data path information'
    )
    parser.add_argument('--bin_path', default='./face/faces_ms1m_112x112/train.rec', type=str,
                        help='path to the binary image file')
    parser.add_argument('--idx_path', default='./face/faces_ms1m_112x112/train.idx', type=str,
                        help='path to the image index path')
    parser.add_argument('--tfrecords_file_path', default='./face/align/', type=str,
                        help='path to the output of tfrecords file path')
    args = parser.parse_args()
    return args


def getPicFromRec(imgidx, imgrec, args):
    # output_path = os.path.join(args.tfrecords_file_path, 'tran.tfrecords')
    sameLabelPicCount = 0
    for i in imgidx:
        img_info = imgrec.read_idx(i)
        header, img = mx.recordio.unpack(img_info)
        label = int(header.label)