Fix for ValueError (expected type INT8) (ultralytics#13341)

Noobtoss · Sep 12, 2024 · 5946ea0 · 5946ea0
1 parent d30aad0
commit 5946ea0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/YOLOv8-OpenCV-int8-tflite-Python/main.py b/examples/YOLOv8-OpenCV-int8-tflite-Python/main.py
@@ -258,7 +258,8 @@ def main(self):
  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()