Spaces:
Sleeping
Sleeping
Commit
·
45249b2
1
Parent(s):
8b3c33c
Update app.py
Browse files
app.py
CHANGED
|
@@ -152,14 +152,15 @@ with gr.Blocks() as demo:
|
|
| 152 |
grayscale_cam = cam(input_tensor = input_img_tensor, targets= targets)
|
| 153 |
grayscale_cam = grayscale_cam[0, :]
|
| 154 |
|
| 155 |
-
renormalized_cam = np.zeros(grayscale_cam.shape, dtype=np.float32)
|
| 156 |
-
new_bboxes = [a[0] for a in sections]
|
| 157 |
|
| 158 |
-
for x1, y1, x2, y2 in new_bboxes:
|
| 159 |
-
|
| 160 |
|
| 161 |
-
renormalized_cam = scale_cam_image(renormalized_cam)
|
| 162 |
-
visualization = show_cam_on_image(input_img_copy, renormalized_cam, use_rgb=False, image_weight=gradcam_opa)
|
|
|
|
| 163 |
|
| 164 |
return (visualization,sections)
|
| 165 |
else:
|
|
|
|
| 152 |
grayscale_cam = cam(input_tensor = input_img_tensor, targets= targets)
|
| 153 |
grayscale_cam = grayscale_cam[0, :]
|
| 154 |
|
| 155 |
+
#renormalized_cam = np.zeros(grayscale_cam.shape, dtype=np.float32)
|
| 156 |
+
#new_bboxes = [a[0] for a in sections]
|
| 157 |
|
| 158 |
+
# for x1, y1, x2, y2 in new_bboxes:
|
| 159 |
+
# renormalized_cam[y1:y2, x1:x2] = scale_cam_image(grayscale_cam[y1:y2, x1:x2].copy())
|
| 160 |
|
| 161 |
+
# renormalized_cam = scale_cam_image(renormalized_cam)
|
| 162 |
+
#visualization = show_cam_on_image(input_img_copy, renormalized_cam, use_rgb=False, image_weight=gradcam_opa)
|
| 163 |
+
visualization = show_cam_on_image(input_img_copy, grayscale_cam, use_rgb=False, image_weight=gradcam_opa)
|
| 164 |
|
| 165 |
return (visualization,sections)
|
| 166 |
else:
|