fix: replace deprecated layer.output_shape with layer.output.shape

frankkramer-lab · Sep 11, 2024 · bb93486 · bb93486
1 parent 4290352
commit bb93486
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/aucmedi/xai/methods/gradcam.py b/aucmedi/xai/methods/gradcam.py
@@ -77,7 +77,7 @@ def find_output_layer(self):
         # Iterate over all layers
         for layer in reversed(self.model.layers):
             # Check to see if the layer has a 4D output -> Return layer
-            if len(layer.output_shape) == 4:
+            if len(layer.output.shape) == 4:
                 return layer.name
         # Otherwise, throw exception
         raise ValueError("Could not find 4D layer. Cannot apply Grad-CAM.")

diff --git a/aucmedi/xai/methods/gradcam_pp.py b/aucmedi/xai/methods/gradcam_pp.py
@@ -73,7 +73,7 @@ def find_output_layer(self):
         # Iterate over all layers
         for layer in reversed(self.model.layers):
             # Check to see if the layer has a 4D output -> Return layer
-            if len(layer.output_shape) == 4:
+            if len(layer.output.shape) == 4:
                 return layer.name
         # Otherwise, throw exception
         raise ValueError("Could not find 4D layer. Cannot apply Grad-CAM++.")