From f42f56870c6201a876f025a423eb5540d7438e83 Mon Sep 17 00:00:00 2001 From: alexander Date: Fri, 16 Jul 2021 11:30:56 +0100 Subject: MLECO-2079 Adding the python KWS example Signed-off-by: Eanna O Cathain Change-Id: Ie1463aaeb5e3cade22df8f560ae99a8e1c4a9c17 --- .../examples/speech_recognition/tests/test_decoder.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) (limited to 'python/pyarmnn/examples/speech_recognition/tests/test_decoder.py') diff --git a/python/pyarmnn/examples/speech_recognition/tests/test_decoder.py b/python/pyarmnn/examples/speech_recognition/tests/test_decoder.py index 1db71a47b8..14db7f2064 100644 --- a/python/pyarmnn/examples/speech_recognition/tests/test_decoder.py +++ b/python/pyarmnn/examples/speech_recognition/tests/test_decoder.py @@ -5,13 +5,16 @@ import os import numpy as np -from context import common_utils from context import audio_utils +labels = {0: 'a', 1: 'b', 2: 'c', 3: 'd', 4: 'e', 5: 'f', 6: 'g', 7: 'h', 8: 'i', 9: 'j', 10: 'k', 11: 'l', 12: 'm', + 13: 'n', + 14: 'o', 15: 'p', 16: 'q', 17: 'r', 18: 's', 19: 't', 20: 'u', 21: 'v', 22: 'w', 23: 'x', 24: 'y', + 25: 'z', + 26: "'", 27: ' ', 28: '$'} + def test_labels(test_data_folder): - labels_file = os.path.join(test_data_folder, "wav2letter_labels.txt") - labels = common_utils.dict_labels(labels_file) assert len(labels) == 29 assert labels[26] == "\'" assert labels[27] == r" " @@ -19,10 +22,8 @@ def test_labels(test_data_folder): def test_decoder(test_data_folder): - labels_file = os.path.join(test_data_folder, "wav2letter_labels.txt") - labels = common_utils.dict_labels(labels_file) - output_tensor = os.path.join(test_data_folder, "inf_out.npy") + output_tensor = os.path.join(test_data_folder, "inference_output.npy") encoded = np.load(output_tensor) decoded_text = audio_utils.decode(encoded, labels) - assert decoded_text == "and they walkd immediately out of the apartiment by anothe" + assert decoded_text == "my voice is my pass" -- cgit v1.2.1