signal-grad-cam 0.1.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

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/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.dist-info → signal_grad_cam-0.1.2.dist-info}/METADATA

@@ -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.2.dist-info/RECORD

@@ -0,0 +1,9 @@
+signal_grad_cam/__init__.py,sha256=JyFFQebs1Rm5r9FCgE68Ii39lLckW8N_gj-nAL5hq2U,137
+signal_grad_cam/cam_builder.py,sha256=KY4o-JMd3ru-TeWQTzmCk4qXTUA7qeAGl1jPbf-tKQY,66588
+signal_grad_cam/pytorch_cam_builder.py,sha256=3nrFEekl8yLwafZcp0I_Kg3QUgeBKGoPQmRAlh4vrJI,17521
+signal_grad_cam/tensorflow_cam_builder.py,sha256=VdFg5h53KzrzboHOmJ8ZaWvJc01Jvy2DdAvvQjtesSg,16904
+signal_grad_cam-0.1.2.dist-info/LICENSE,sha256=pCSaMipV39klP0dlf75SHw5PTl00_cLlS-EiC-LmOkw,1088
+signal_grad_cam-0.1.2.dist-info/METADATA,sha256=_twEITZPMrlZ94ZznndbcWbpaCq8fw6k4k44tJDpelo,10986
+signal_grad_cam-0.1.2.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+signal_grad_cam-0.1.2.dist-info/top_level.txt,sha256=S6lf3mfh2uGXJKnUS3qnw6arQu-x3gO8m82WdY7JAIA,16
+signal_grad_cam-0.1.2.dist-info/RECORD,,

signal_grad_cam-0.1.1.dist-info/RECORD

@@ -1,9 +0,0 @@
-signal_grad_cam/__init__.py,sha256=JyFFQebs1Rm5r9FCgE68Ii39lLckW8N_gj-nAL5hq2U,137
-signal_grad_cam/cam_builder.py,sha256=KY4o-JMd3ru-TeWQTzmCk4qXTUA7qeAGl1jPbf-tKQY,66588
-signal_grad_cam/pytorch_cam_builder.py,sha256=JPoeTk_fIIg0MtfIA3RuEzNtTcCfaHdzpaSJNwXHCGc,16566
-signal_grad_cam/tensorflow_cam_builder.py,sha256=RCokRq6Q0ZlOphCHXHkRdjcAEy969Jd8vNxtYc6Sq_o,15890
-signal_grad_cam-0.1.1.dist-info/LICENSE,sha256=pCSaMipV39klP0dlf75SHw5PTl00_cLlS-EiC-LmOkw,1088
-signal_grad_cam-0.1.1.dist-info/METADATA,sha256=eZaIXadKb4hoBlZgfRueUNichVHC-YtXGYMpLmZ_0Xo,10986
-signal_grad_cam-0.1.1.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-signal_grad_cam-0.1.1.dist-info/top_level.txt,sha256=S6lf3mfh2uGXJKnUS3qnw6arQu-x3gO8m82WdY7JAIA,16
-signal_grad_cam-0.1.1.dist-info/RECORD,,