experiments/STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Some code beautification

Browse Source
This commit is contained in:
Tilman Kamp 2019-04-02 12:15:08 +02:00
parent 5645285d25
commit 8e78e17904
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
bin/import_cv2.py
View File

@ -135,8 +135,7 @@ if __name__ == "__main__":
parser.add_argument('tsv_dir', help='Directory containing tsv files') parser.add_argument('tsv_dir', help='Directory containing tsv files')
parser.add_argument('--audio_dir', help='Directory containing the audio clips - defaults to "<tsv_dir>/clips"') parser.add_argument('--audio_dir', help='Directory containing the audio clips - defaults to "<tsv_dir>/clips"')
parser.add_argument('--filter_alphabet', help='Exclude samples with characters not in provided alphabet') parser.add_argument('--filter_alphabet', help='Exclude samples with characters not in provided alphabet')
parser.add_argument('--normalize', action='store_true', help='Converts diacritic characters to their base ones') parser.add_argument('--normalize', default=False, action='store_true', help='Converts diacritic characters to their base ones')
parser.set_defaults(normalize=False)
params = parser.parse_args() params = parser.parse_args()
audio_dir = params.audio_dir if params.audio_dir else os.path.join(params.tsv_dir, 'clips') audio_dir = params.audio_dir if params.audio_dir else os.path.join(params.tsv_dir, 'clips')
@ -152,7 +151,7 @@ if __name__ == "__main__":
try: try:
[alphabet.label_from_string(c) for c in label] [alphabet.label_from_string(c) for c in label]
except KeyError: except KeyError:
return None label = None
return label return label
_preprocess_data(params.tsv_dir, audio_dir, label_filter) _preprocess_data(params.tsv_dir, audio_dir, label_filter)