From 97ddc06e52fbcabfd8ede7a00e9494c663186b92 Mon Sep 17 00:00:00 2001 From: Raviv Shalev Date: Tue, 7 Dec 2021 15:18:09 +0200 Subject: MLECO-2493 Add python OD example with TFLite delegate Signed-off-by: Raviv Shalev Change-Id: I25fcccbf912be0c5bd4fbfd2e97552341958af35 --- python/pyarmnn/examples/speech_recognition/run_audio_file.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'python/pyarmnn/examples/speech_recognition/run_audio_file.py') diff --git a/python/pyarmnn/examples/speech_recognition/run_audio_file.py b/python/pyarmnn/examples/speech_recognition/run_audio_file.py index 0430f68c16..ddf6cb704c 100644 --- a/python/pyarmnn/examples/speech_recognition/run_audio_file.py +++ b/python/pyarmnn/examples/speech_recognition/run_audio_file.py @@ -12,7 +12,7 @@ sys.path.insert(1, os.path.join(script_dir, '..', 'common')) from argparse import ArgumentParser from network_executor import ArmnnNetworkExecutor -from utils import prepare_input_tensors +from utils import prepare_input_data from audio_capture import AudioCaptureParams, capture_audio from audio_utils import decode_text, display_text from wav2letter_mfcc import Wav2LetterMFCC, W2LAudioPreprocessor @@ -78,10 +78,11 @@ def main(args): print("Processing Audio Frames...") for audio_data in buffer: # Prepare the input Tensors - input_tensors = prepare_input_tensors(audio_data, network.input_binding_info, preprocessor) + input_data = prepare_input_data(audio_data, network.get_data_type(), network.get_input_quantization_scale(0), + network.get_input_quantization_offset(0), preprocessor) # Run inference - output_result = network.run(input_tensors) + output_result = network.run([input_data]) # Slice and Decode the text, and store the right context current_r_context, text = decode_text(is_first_window, labels, output_result) -- cgit v1.2.1