fix: Fix visualizer callback showing wrong heatmaps after normalizati…

…on score refactoring (#121) * fix: Fix visualizer callback showing wrong heatmaps after normalization score refactoring * build: Upgrade version, update changelog
orobix · Jun 5, 2024 · 44c0a4f · 44c0a4f
1 parent 922621d
commit 44c0a4f
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,6 +2,12 @@
 # Changelog
 All notable changes to this project will be documented in this file.
 
+### [2.1.10]
+
+#### Fixed
+
+- Fix anomaly visualizer callback showing wrong heatmaps after anomaly score refactoring
+
 ### [2.1.9]
 
 #### Updated

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "quadra"
-version = "2.1.9"
+version = "2.1.10"
 description = "Deep Learning experiment orchestration library"
 authors = [
 	"Federico Belotti <[email protected]>",

diff --git a/quadra/__init__.py b/quadra/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "2.1.9"
+__version__ = "2.1.10"
 
 
 def get_version():

diff --git a/quadra/callbacks/anomalib.py b/quadra/callbacks/anomalib.py
@@ -22,6 +22,8 @@
 from skimage.segmentation import mark_boundaries
 from tqdm import tqdm
 
+from quadra.utils.anomaly import MapOrValue
+
 
 class Visualizer:
     """Anomaly Visualization.
@@ -204,14 +206,20 @@ def on_test_batch_end(
             denormalized_image = Denormalize()(image.cpu())
             current_true_mask = true_mask.cpu().numpy()
             current_anomaly_map = anomaly_map.cpu().numpy()
+            # Normalize the map and rescale it to 0-1 range
+            # In this case we are saying that the anomaly map is in the range [normalized_th - 50, normalized_th + 50]
+            # This allow to have a stronger color for the anomalies and a lighter one for really normal regions
+            # It's also independent from the max or min anomaly score!
+            normalized_map: MapOrValue = (current_anomaly_map - (threshold - 50)) / 100
+            normalized_map = np.clip(normalized_map, 0, 1)
 
             output_label_folder = "ok" if pred_label == gt_label else "wrong"
 
             if self.plot_only_wrong and output_label_folder == "ok":
                 continue
 
             heatmap = superimpose_anomaly_map(
-                current_anomaly_map, denormalized_image, normalize=not self.inputs_are_normalized
+                normalized_map, denormalized_image, normalize=not self.inputs_are_normalized
             )
 
             if isinstance(threshold, float):