diff --git a/speech2text/accuracy_eval.py b/speech2text/accuracy_eval.py index 3c7466ff8f..eb6cc53299 100644 --- a/speech2text/accuracy_eval.py +++ b/speech2text/accuracy_eval.py @@ -28,6 +28,7 @@ max_duration = float(os.environ.get("MAX_DURATION", "30.0")) + labels = [ " ", "a", @@ -56,7 +57,22 @@ "x", "y", "z", - "'"] + "'", + "0", + "1", + "2", + "3", + "4", + "5", + "6", + "7", + "8", + "9", + "$", + "%", + "+", + "-"] + dtype_map = { "int8": 'b', "int16": 'h', diff --git a/speech2text/reference_SUT.py b/speech2text/reference_SUT.py index fcd5385507..63d491a00f 100644 --- a/speech2text/reference_SUT.py +++ b/speech2text/reference_SUT.py @@ -90,7 +90,22 @@ def get_start_cores(start_cores="0"): "x", "y", "z", - "'"] + "'", + "0", + "1", + "2", + "3", + "4", + "5", + "6", + "7", + "8", + "9", + "$", + "%", + "+", + "-"] + labels_dict = {} for i in range(len(labels)): labels_dict[labels[i]] = i