stevfoy commited on
Commit
54fc56a
·
1 Parent(s): aea11dd
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -83,7 +83,7 @@ def process_image_and_plot(image,iou_threshold=0.5, threshold=0.4):
83
  cam = YoloCAM(model=model, target_layers=[model.model.layers[-2]], use_cuda=False)
84
 
85
  grayscale_cam = cam(transformed_image, scaled_anchors)[0, :, :]
86
- img = cv2.resize(image, (416, 416))
87
  img = np.float32(img) / 255
88
  cam_image = show_cam_on_image(img, grayscale_cam, use_rgb=True)
89
 
 
83
  cam = YoloCAM(model=model, target_layers=[model.model.layers[-2]], use_cuda=False)
84
 
85
  grayscale_cam = cam(transformed_image, scaled_anchors)[0, :, :]
86
+ img = cv2.resize(image, (608, 608))
87
  img = np.float32(img) / 255
88
  cam_image = show_cam_on_image(img, grayscale_cam, use_rgb=True)
89