extra

app.py

@@ -69,7 +69,21 @@ def process_image_and_plot(image,iou_threshold=0.5, threshold=0.4):
     transformed_image = transforms(image=image)["image"].unsqueeze(0)
     print("transformed_image 2") 
     output = model(transformed_image)
-    print("transformed_image 3") 
+    
+    
+   
+
+    labels = []
+    sample_metrics = []  # List of tuples (TP, confs, pred)
+
+	# Extract labels
+    
+
+    with torch.no_grad():
+        outputs = model(imgs)
+        outputs = non_max_suppression(outputs, conf_thres=iou_threshold, iou_thres=threshold)
+        # sample_metrics = get_batch_statistics(outputs, targets, iou_threshold=threshold)
+
     bboxes = [[] for _ in range(1)]
     
     print("transformed_image 4") 
@@ -81,9 +95,9 @@ def process_image_and_plot(image,iou_threshold=0.5, threshold=0.4):
         for idx, box in enumerate(boxes_scale_i):
             bboxes[idx] += box
     print("transformed_image 5") 
-    nms_boxes = non_max_suppression(
-        bboxes[0], iou_threshold=iou_threshold, threshold=threshold, box_format="midpoint",
-    )
+    
+
+    
     print("transformed_image 6") 
     fig = plot_image(transformed_image[0].permute(1, 2, 0), nms_boxes)
     print("transformed_image 7")