signal-grad-cam 0.1.1__tar.gz → 0.1.2__tar.gz

{signal_grad_cam-0.1.1 → signal_grad_cam-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: signal_grad_cam
-Version: 0.1.1
+Version: 0.1.2
 Summary: SignalGrad-CAM aims at generalising Grad-CAM to one-dimensional applications, while enhancing usability and efficiency.
 Home-page: https://github.com/samuelepe11/signal_grad_cam
 Author: Samuele Pe

{signal_grad_cam-0.1.1 → signal_grad_cam-0.1.2}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as f:
 
 setup(
     name="signal_grad_cam",
-    version="0.1.1",
+    version="0.1.2",
     description="SignalGrad-CAM aims at generalising Grad-CAM to one-dimensional applications, while enhancing usability"
                 " and efficiency.",
     keywords="XAI, class activation maps, CNN, time series",

{signal_grad_cam-0.1.1 → signal_grad_cam-0.1.2}/signal_grad_cam/pytorch_cam_builder.py

@@ -183,9 +183,26 @@ class TorchCamBuilder(CamBuilder):
             # during derivation
             target_scores = torch.log(outputs)
             target_probs = outputs
+
+            # Adjust results for binary network
+            if len(outputs.shape) == 1:
+                target_scores = torch.stack([-target_scores, target_scores], dim=1)
+                target_probs = torch.stack([1 - target_probs, target_probs], dim=1)
+            elif len(outputs.shape) == 2 and outputs.shape[1] == 1:
+                target_scores = torch.cat([-target_scores, target_scores], dim=1)
+                target_probs = torch.cat([1 - target_probs, target_probs], dim=1)
         else:
+            if len(outputs.shape) == 1:
+                outputs = torch.stack([-outputs, outputs], dim=1)
+            elif len(outputs.shape) == 2 and outputs.shape[1] == 1:
+                outputs = torch.cat([-outputs, outputs], dim=1)
             target_scores = outputs
-            target_probs = torch.softmax(target_scores, dim=1)
+
+            if len(outputs.shape) == 2 and outputs.shape[1] > 1:
+                target_probs = torch.softmax(target_scores, dim=1)
+            else:
+                tmp = torch.sigmoid(target_scores[:, 1])
+                target_probs = torch.stack([1 - tmp, tmp], dim=1)
 
         target_probs = target_probs[:, target_class].cpu().detach().numpy()
 

{signal_grad_cam-0.1.1 → signal_grad_cam-0.1.2}/signal_grad_cam/tensorflow_cam_builder.py

@@ -189,7 +189,7 @@ class TfCamBuilder(CamBuilder):
 
         grad_model = keras.models.Model(self.model.inputs, [target_layer.output, self.model.output])
         extra_inputs_list = extra_inputs_list or []
-        with tf.GradientTape() as tape:
+        with (tf.GradientTape() as tape):
             self.activations, outputs = grad_model([data_batch] + extra_inputs_list)
 
             if softmax_final:
@@ -197,9 +197,26 @@ class TfCamBuilder(CamBuilder):
                 # during derivation
                 target_scores = tf.math.log(outputs)
                 target_probs = outputs
+
+                # Adjust results for binary network
+                if len(outputs.shape) == 1:
+                    target_scores = tf.stack([-target_scores, target_scores], axis=1)
+                    target_probs = tf.stack([1 - target_probs, target_probs], axis=1)
+                elif len(outputs.shape) == 2 and outputs.shape[1] == 1:
+                    target_scores = tf.concat([-target_scores, target_scores], axis=1)
+                    target_probs = tf.concat([1 - target_probs, target_probs], axis=1)
             else:
+                if len(outputs.shape) == 1:
+                    outputs = tf.stack([-outputs, outputs], axis=1)
+                elif len(outputs.shape) == 2 and outputs.shape[1] == 1:
+                    outputs = tf.concat([-outputs, outputs], axis=1)
                 target_scores = outputs
-                target_probs = tf.nn.softmax(target_scores, axis=1)
+
+                if len(outputs.shape) == 2 and outputs.shape[1] > 1:
+                    target_probs = tf.nn.softmax(target_scores, axis=1)
+                else:
+                    tmp = tf.math.sigmoid(target_scores[:, 1])
+                    target_probs = tf.stack([1 - tmp, tmp], axis=1)
 
             target_scores = target_scores[:, target_class]
             target_probs = target_probs[:, target_class]

{signal_grad_cam-0.1.1 → signal_grad_cam-0.1.2}/signal_grad_cam.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: signal-grad-cam
-Version: 0.1.1
+Version: 0.1.2
 Summary: SignalGrad-CAM aims at generalising Grad-CAM to one-dimensional applications, while enhancing usability and efficiency.
 Home-page: https://github.com/samuelepe11/signal_grad_cam
 Author: Samuele Pe