We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent af2d53a commit 21ef872Copy full SHA for 21ef872
pytorch_grad_cam/score_cam.py
@@ -37,7 +37,7 @@ def get_cam_weights(self,
37
upsampled.size(1), -1).min(dim=-1)[0]
38
39
maxs, mins = maxs[:, :, None, None], mins[:, :, None, None]
40
- upsampled = (upsampled - mins) / (maxs - mins)
+ upsampled = (upsampled - mins) / (maxs - mins + 1e-8)
41
42
input_tensors = input_tensor[:, None,
43
:, :] * upsampled[:, :, None, :, :]
0 commit comments