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

Make evaluate_tflite.py work with relative paths in the CSV

Browse Source
This commit is contained in:
Reuben Morais 2020-01-12 10:59:30 +01:00
parent 3df20fee52
commit 42fe30d572
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
evaluate_tflite.py
View File

@ -93,6 +93,9 @@ def main():
count = 0
for row in csvreader:
count += 1
# Relative paths are relative to the folder the CSV file is in
if not os.path.isabs(row['wav_filename']):
row['wav_filename'] = os.path.join(os.path.dirname(args.csv), row['wav_filename'])
work_todo.put({'filename': row['wav_filename'], 'transcript': row['transcript']})
wav_filenames.extend(row['wav_filename'])