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

Output full paths to results dump in evaluate_tflite.py

Browse Source
This commit is contained in:
Reuben Morais 2020-01-12 11:01:54 +01:00
parent 33e725bb25
commit c28f61d370
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
evaluate_tflite.py
View File

@ -44,14 +44,13 @@ def tflite_worker(model, lm, trie, queue_in, queue_out, gpu_mask):
msg = queue_in.get()
filename = msg['filename']
wavname = os.path.splitext(os.path.basename(filename))[0]
fin = wave.open(filename, 'rb')
audio = np.frombuffer(fin.readframes(fin.getnframes()), np.int16)
fin.close()
decoded = ds.stt(audio)
queue_out.put({'wav': wavname, 'prediction': decoded, 'ground_truth': msg['transcript']})
queue_out.put({'wav': filename, 'prediction': decoded, 'ground_truth': msg['transcript']})
except FileNotFoundError as ex:
print('FileNotFoundError: ', ex)