python 制作录音文件

Posted 2021-05-08

import mxnet as mx
import os
import os.path as osp
import tqdm
import argparse
import cv2

parse = argparse.ArgumentParser(description='Image to rec')
parse.add_argument('--data_dir', help='Dataset root')
parse.add_argument('--output', help='Output dir')
parse.add_argument('--type', default='train', type=str, help='train or val ot test type')
args = parse.parse_args()


_idx = osp.join(args.output, '{}.idx'.format(args.type))
_rec = osp.join(args.output, '{}.rec'.format(args.type))
rec = mx.recordio.MXIndexedRecordIO(_idx, _rec, 'w')

img_files = os.listdir(args.data_dir)
total = len(img_files)

print('total {} files'.format(total))

cnt = 0
with tqdm.tqdm(iterable=range(total), total=total, ncols=0, unit=' items') as t:
    for id, i in enumerate(t):
        f = img_files[id]
        vid_id = f.split('_')[-1] # video id
        img_names = os.listdir(osp.join(args.data_dir,f))
        for img_name in img_names:
            header = mx.recordio.IRHeader(0, vid_id, cnt, 0)
            img=cv2.imread(osp.join(args.data_dir, f, img_name))
            packed = mx.recordio.pack_img(header, img)
            rec.write_idx(cnt, packed)
            cnt+=1