fix(XAI): added missing inverse of occlusion sensitivity heatmap

frankkramer-lab · Jul 8, 2022 · 05169cf · 05169cf
1 parent 4711894
commit 05169cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aucmedi/xai/methods/occlusion_sensitivity.py b/aucmedi/xai/methods/occlusion_sensitivity.py
@@ -91,7 +91,7 @@ def compute_heatmap(self, image, class_index, eps=1e-8):
                 sensitivity_map[
                     top_left_y:top_left_y + self.patch_size,
                     top_left_x:top_left_x + self.patch_size,
-                ] = confidence
+                ] = 1 - confidence
         # Return the resulting sensitivity map (automatically a heatmap)
         return sensitivity_map