Update common_detection.py

common_detection.py

@@ -1,4 +1,3 @@
-# common_detection.py
 import cv2
 import numpy as np
 from PIL import Image
@@ -21,9 +20,13 @@ def perform_detection(image, interpreter, labels, input_details, output_details,
     classes = interpreter.get_tensor(output_details[1]['index'])[0]
     scores = interpreter.get_tensor(output_details[2]['index'])[0]
 
+    print(f"boxes shape: {boxes.shape}")
+    print(f"classes shape: {classes.shape}")
+    print(f"scores shape: {scores.shape}")
+
     detections = []
     for i in range(len(scores)):
-        if (scores[i] > 0.5):
+        if scores[i] > 0.5:
             ymin = int(max(1, (boxes[i][0] * imH)))
             xmin = int(max(1, (boxes[i][1] * imW)))
             ymax = int(min(imH, (boxes[i][2] * imH)))