MLECO-2079 Adding the python KWS example
Signed-off-by: Eanna O Cathain <eanna.ocathain@arm.com>
Change-Id: Ie1463aaeb5e3cade22df8f560ae99a8e1c4a9c17
diff --git a/python/pyarmnn/examples/speech_recognition/tests/test_decoder.py b/python/pyarmnn/examples/speech_recognition/tests/test_decoder.py
index 1db71a4..14db7f2 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 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_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"