36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
#!/usr/bin/env python
|
|
import os
|
|
import argparse
|
|
|
|
CLASSES = 'unknown, yes, no, up, down, left, right, on, off, stop, go'.split(
|
|
', ')
|
|
CLASS_TO_IDX = {CLASSES[i]: str(i) for i in range(len(CLASSES))}
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(
|
|
description='prepare kaldi format file for google speech command')
|
|
parser.add_argument(
|
|
'--wav_list',
|
|
required=True,
|
|
help='full path of a wav file in google speech command dataset')
|
|
parser.add_argument('--data_dir',
|
|
required=True,
|
|
help='folder to write kaldi format files')
|
|
args = parser.parse_args()
|
|
|
|
data_dir = args.data_dir
|
|
f_wav_scp = open(os.path.join(data_dir, 'wav.scp'), 'w')
|
|
f_text = open(os.path.join(data_dir, 'text'), 'w')
|
|
with open(args.wav_list) as f:
|
|
for line in f.readlines():
|
|
keyword, file_name = line.strip().split('/')[-2:]
|
|
file_name_new = file_name.split('.')[0]
|
|
wav_id = '_'.join([keyword, file_name_new])
|
|
file_dir = line.strip()
|
|
f_wav_scp.writelines(wav_id + ' ' + file_dir + '\n')
|
|
label = CLASS_TO_IDX[
|
|
keyword] if keyword in CLASS_TO_IDX else CLASS_TO_IDX["unknown"]
|
|
f_text.writelines(wav_id + ' ' + str(label) + '\n')
|
|
f_wav_scp.close()
|
|
f_text.close()
|