add a small constant to avoid division by zero (#449)

jacobgil · Oct 6, 2023 · 21ef872 · 21ef872
1 parent af2d53a
commit 21ef872
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_grad_cam/score_cam.py b/pytorch_grad_cam/score_cam.py
@@ -37,7 +37,7 @@ def get_cam_weights(self,
  upsampled.size(1), -1).min(dim=-1)[0]
 
  maxs, mins = maxs[:, :, None, None], mins[:, :, None, None]
- upsampled = (upsampled - mins) / (maxs - mins)
+ upsampled = (upsampled - mins) / (maxs - mins + 1e-8)
 
  input_tensors = input_tensor[:, None,
  :, :] * upsampled[:, :, None, :, :]