Update display.py

display.py

@@ -6,7 +6,7 @@ import random
 from albumentations.pytorch import ToTensorV2
 from yolov3 import YOLOV3_PL
 
-def inference(image: np.ndarray, iou_thresh: float = 0.75, thresh: float = 0.75, show_cam: bool = False, transparency: float = 0.5):
+def inference(image: np.ndarray, iou_thresh: float = 0.75, thresh: float = 0.75,  transparency: float = 0.5):
     model = YOLOV3_PL()
     model.load_state_dict(torch.load("model.pth", map_location=torch.device('cpu')), strict=False)
 
@@ -39,14 +39,8 @@ def inference(image: np.ndarray, iou_thresh: float = 0.75, thresh: float = 0.75,
         bboxes[0], iou_threshold=iou_thresh, threshold=thresh, box_format="midpoint",
     )
     plot_img = draw_predictions(image, nms_boxes, class_labels=config.PASCAL_CLASSES)
-    if not show_cam:
-        return [plot_img]
+    return [plot_img]
 
-    grayscale_cam = cam(transformed_image, scaled_anchors)[0, :, :]
-    img = cv2.resize(image, (416, 416))
-    img = np.float32(img) / 255
-    cam_image = show_cam_on_image(img, grayscale_cam, use_rgb=True, image_weight=transparency)
-    return [plot_img, cam_image]
 
 
 def draw_predictions(image: np.ndarray, boxes: list[list], class_labels: list[str]) -> np.ndarray: