Show visualize heatmap

[vvthai10]

I want to view the heatmap like in the paper, But I can't find the way. Someone can help me?. Give me some code or document to do it

[LengSenghak]

To create a heatmap similar to the one in the paper, you can use the following code snippet:

Assuming you have the test image variable called test_img in numpy array and anomaly map with score from 0-1. You can follow these code snippet.

Code

import numpy as np
import cv2
from scipy.ndimage import gaussian_filter

# Assume anomaly_map is a 2D array with values between 0 and 1
anomaly_map = gaussian_filter(anomaly_map, sigma=4)  # Apply Gaussian filter to smooth the map
anomaly_map_8bit = (anomaly_map * 255).astype(np.uint8)  # Convert to 8-bit unsigned integer

# Apply a color map to the anomaly score map
heat_map = cv2.applyColorMap(anomaly_map_8bit, cv2.COLORMAP_JET)

# Resize the heatmap to match the original image size
heat_map = cv2.resize(heat_map, (test_img.shape[1], test_img.shape[0]))

# Overlay the heatmap on the original image
output_img = cv2.addWeighted(heat_map, 0.5, cv2.cvtColor(test_img, cv2.COLOR_RGB2BGR), 0.7, 0)

Hope this help