AkashDataScience commited on
Commit
02b3e92
·
1 Parent(s): a365161

Changed image

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -122,8 +122,8 @@ def inference(input_img, conf_thres, iou_thres, is_false_detection_images=True,
122
  misclassified_images = None
123
 
124
  cam = EigenCAM(model, target_layers)
125
- grayscale_cam = cam(rgb_img)[0, :, :]
126
- cam_image = show_cam_on_image(im_resized, grayscale_cam, use_rgb=True)
127
 
128
  return im0, cam_image, misclassified_images
129
 
 
122
  misclassified_images = None
123
 
124
  cam = EigenCAM(model, target_layers)
125
+ grayscale_cam = cam(im_resized)[0, :, :]
126
+ cam_image = show_cam_on_image(rgb_img, grayscale_cam, use_rgb=True)
127
 
128
  return im0, cam_image, misclassified_images
129