From cfc85998e472f86f6a8f9e325fb4b140222f20ac Mon Sep 17 00:00:00 2001
From: "michael.divia" <michael.divia@etu.hesge.ch>
Date: Wed, 9 Apr 2025 21:39:46 +0200
Subject: [PATCH] Problem with sizes

---
 python/pokedex_rpi.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/python/pokedex_rpi.py b/python/pokedex_rpi.py
index 34dca02..0110140 100644
--- a/python/pokedex_rpi.py
+++ b/python/pokedex_rpi.py
@@ -29,7 +29,11 @@ with open(json_path, "r") as f:
 
 # --- Hailo Inference with PiCamera2 ---
 with Hailo(hef_path) as hailo:
-    print(f"-- Model input shape: {input_shape}")
+    model_h, model_w, model_c = hailo.get_input_shape()
+    print(f"-- Model input shape from Hailo: {(model_h, model_w, model_c)}")
+    print(f"-- Input format: {hailo.get_input_format()}")
+    print(f"-- Expected input buffer size: {hailo.get_input_frame_size()}")
+
 
     picam2 = Picamera2()
     config = picam2.create_still_configuration(
-- 
GitLab