From 0dac8b66fdc5c2320bf9c9ab52e9db0290a44c2d Mon Sep 17 00:00:00 2001
From: Dick Ameln <dick.ameln@intel.com>
Date: Tue, 21 Dec 2021 22:26:31 +0100
Subject: [PATCH] fix heatmap color scheme
---
anomalib/utils/post_process.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/anomalib/utils/post_process.py b/anomalib/utils/post_process.py
index 18ffea2a24..246ba6b2b3 100644
--- a/anomalib/utils/post_process.py
+++ b/anomalib/utils/post_process.py
@@ -37,6 +37,7 @@ def anomaly_map_to_color_map(anomaly_map: np.ndarray, normalize: bool = True) ->
anomaly_map = anomaly_map.astype(np.uint8)
anomaly_map = cv2.applyColorMap(anomaly_map, cv2.COLORMAP_JET)
+ anomaly_map = cv2.cvtColor(anomaly_map, cv2.COLOR_BGR2RGB)
return anomaly_map
@@ -64,7 +65,6 @@ def superimpose_anomaly_map(
anomaly_map = anomaly_map_to_color_map(anomaly_map.squeeze(), normalize=normalize)
superimposed_map = cv2.addWeighted(anomaly_map, alpha, image, (1 - alpha), gamma)
- superimposed_map = cv2.cvtColor(superimposed_map, cv2.COLOR_BGR2RGB)
return superimposed_map