| 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() |
|
|
