extra

app.py

@@ -86,7 +86,9 @@ def process_image_and_plot(image,iou_threshold=0.5, threshold=0.4):
         boxes = [[] for _ in range(1)]
     
         for i in range(3):
+            print("output[i].shape", output[i].shape)
             batch_size, A, S, _, _ = output[i].shape
+            
             anchor = scaled_anchors[i]
             boxes_scale_i = cells_to_bboxes(output[i], anchor, S=S, is_preds=True)
             for idx, box in enumerate(boxes_scale_i):