Spaces:
Sleeping
Sleeping
Commit
·
02b3e92
1
Parent(s):
a365161
Changed image
Browse files
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(
|
126 |
-
cam_image = show_cam_on_image(
|
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 |
|