From cf081c4db7972a4fe42794a234d867ce0930b4b2 Mon Sep 17 00:00:00 2001
From: czy10383 <81451461+czy10383@users.noreply.github.com>
Date: Tue, 4 Jun 2024 18:23:18 +0800
Subject: [PATCH] Fix for ValueError (expected type INT8) (#13341)

---
 examples/YOLOv8-OpenCV-int8-tflite-Python/main.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/examples/YOLOv8-OpenCV-int8-tflite-Python/main.py b/examples/YOLOv8-OpenCV-int8-tflite-Python/main.py
index 0a08756b..1e3a8e99 100644
--- a/examples/YOLOv8-OpenCV-int8-tflite-Python/main.py
+++ b/examples/YOLOv8-OpenCV-int8-tflite-Python/main.py
@@ -258,7 +258,8 @@ class Yolov8TFLite:
         img_data = img_data.transpose((0, 2, 3, 1))
 
         scale, zero_point = input_details[0]["quantization"]
-        interpreter.set_tensor(input_details[0]["index"], img_data)
+        img_data_int8 = (img_data / scale + zero_point).astype(np.int8)
+        interpreter.set_tensor(input_details[0]["index"], img_data_int8)
 
         # Run inference
         interpreter.invoke()