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), BBOX)
+    fig = plot_image(transformed_image[0].permute(1, 2, 0), detect)
     # cam = YoloCAM(model=model, target_layers=[model.model.layers[-2]], use_cuda=False)
     
     # grayscale_cam = cam(transformed_image, scaled_anchors)[0, :, :]

utils_for_app.py

@@ -186,7 +186,7 @@ def plot_image(image, boxes):
             verticalalignment="top",
             bbox={"color": colors[int(class_pred)], "pad": 0},
         )
-        print("found ",box ," ",colors[int(class_pred)]," ", upper_left_x ," ", width, " ",upper_left_y ," ", height," ",upper_left_x ," ", width," ", upper_left_y ," ", height)
+        print("found ",box )
 
     plt.show()
     return fig