extra

app.py

@@ -84,7 +84,7 @@ def process_image_and_plot(image,iou_threshold=0.5, threshold=0.4):
         BBOX = rescale_boxes(detect[0], img_size, image.shape[:2])
     
 
-    fig = plot_image(transformed_image[0].permute(1, 2, 0), detect)
+    fig = plot_image(transformed_image[0].permute(1, 2, 0), detect[0])
     # cam = YoloCAM(model=model, target_layers=[model.model.layers[-2]], use_cuda=False)
     
     # grayscale_cam = cam(transformed_image, scaled_anchors)[0, :, :]