From b49c563b99e8d1a0a250c09d27fdba5d3b034394 Mon Sep 17 00:00:00 2001 From: Dick Ameln Date: Wed, 22 Dec 2021 08:59:09 +0100 Subject: [PATCH] Fix heatmap color scheme (#44) This is to fix heatmap color map to RGB by default. --- 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