MLECO-2493 Add python OD example with TFLite delegate
Signed-off-by: Raviv Shalev <raviv.shalev@arm.com>
Change-Id: I25fcccbf912be0c5bd4fbfd2e97552341958af35
diff --git a/python/pyarmnn/examples/tests/test_common_utils.py b/python/pyarmnn/examples/tests/test_common_utils.py
index 28d68ea..254eba6 100644
--- a/python/pyarmnn/examples/tests/test_common_utils.py
+++ b/python/pyarmnn/examples/tests/test_common_utils.py
@@ -2,9 +2,13 @@
# SPDX-License-Identifier: MIT
import os
+import time
+import cv2
+import numpy as np
from context import cv_utils
from context import utils
+from utils import Profiling
def test_get_source_encoding(test_data_folder):
@@ -17,3 +21,22 @@
label_file = os.path.join(test_data_folder, "labelmap.txt")
labels_map = utils.dict_labels(label_file)
assert labels_map is not None
+
+
+def test_preprocess(test_data_folder):
+ content_image = "messi5.jpg"
+ target_shape = (1, 256, 256, 3)
+ padding = True
+ image = cv2.imread(os.path.join(test_data_folder, content_image))
+ image = cv_utils.preprocess(image, np.float32, target_shape, True, padding)
+
+ assert image.shape == target_shape
+
+
+def test_profiling():
+ profiler = Profiling(True)
+ profiler.profiling_start()
+ time.sleep(1)
+ period = profiler.profiling_stop_and_print_us("Sleep for 1 second")
+ assert (1_000_000 < period < 1_002_000)
+