Binary model target

jacobgil · jacobgil · commit 58a565a48fb3 · 2023-06-15T15:08:03.000+03:00
diff --git a/pytorch_grad_cam/utils/model_targets.py b/pytorch_grad_cam/utils/model_targets.py
@@ -32,7 +32,7 @@ def __call__(self, model_output):
             sign = 1
         else:
             sign = -1
-        return model_output * sign
+        return torch.abs(model_output) * sign
 
 
 class SoftmaxOutputTarget:
