sanjanatule commited on
Commit
45249b2
·
1 Parent(s): 8b3c33c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -6
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
- 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
 
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: