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 |
|