54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
import argparse
|
|
import logging
|
|
import glob
|
|
import os
|
|
from tqdm import tqdm
|
|
|
|
|
|
logger = logging.getLogger()
|
|
|
|
|
|
def main():
|
|
wavs = list(glob.glob('{}/*/*/*wav'.format(args.src_dir)))
|
|
|
|
utt2wav, utt2text, utt2spk, spk2utt = {}, {}, {}, {}
|
|
for wav in tqdm(wavs):
|
|
txt = wav.replace('.wav', '.normalized.txt')
|
|
if not os.path.exists(txt):
|
|
logger.warning('{} do not exsist'.format(txt))
|
|
continue
|
|
with open(txt) as f:
|
|
content = ''.join(l.replace('\n', '') for l in f.readline())
|
|
utt = os.path.basename(wav).replace('.wav', '')
|
|
spk = utt.split('_')[0]
|
|
utt2wav[utt] = wav
|
|
utt2text[utt] = content
|
|
utt2spk[utt] = spk
|
|
if spk not in spk2utt:
|
|
spk2utt[spk] = []
|
|
spk2utt[spk].append(utt)
|
|
|
|
with open('{}/wav.scp'.format(args.des_dir), 'w') as f:
|
|
for k, v in utt2wav.items():
|
|
f.write('{} {}\n'.format(k, v))
|
|
with open('{}/text'.format(args.des_dir), 'w') as f:
|
|
for k, v in utt2text.items():
|
|
f.write('{} {}\n'.format(k, v))
|
|
with open('{}/utt2spk'.format(args.des_dir), 'w') as f:
|
|
for k, v in utt2spk.items():
|
|
f.write('{} {}\n'.format(k, v))
|
|
with open('{}/spk2utt'.format(args.des_dir), 'w') as f:
|
|
for k, v in spk2utt.items():
|
|
f.write('{} {}\n'.format(k, ' '.join(v)))
|
|
return
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--src_dir',
|
|
type=str)
|
|
parser.add_argument('--des_dir',
|
|
type=str)
|
|
args = parser.parse_args()
|
|
main()
|