Update prepare_speech_command.py

This commit is contained in:
xiaohou 2021-11-18 16:14:31 +08:00 committed by GitHub
parent a7f54824a3
commit a73f53d4aa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -7,14 +7,11 @@ 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))} CLASS_TO_IDX = {CLASSES[i]: str(i) for i in range(len(CLASSES))}
if __name__ == '__main__': if __name__ == '__main__':
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(description='prepare kaldi format file')
description=
'prepare kaldi format file for google speech command dataset ')
parser.add_argument( parser.add_argument(
'--wav_list', '--wav_list',
required=True, required=True,
help= help='wave_list containts full path of a wav file'
'wave list is a file containts full path of a wav file in google speech command dataset'
) )
parser.add_argument('--data_dir', parser.add_argument('--data_dir',
required=True, required=True,