How to visualize attention map

Hi all.
I have an image and its corresponding attention map, which is a [1, H, W] tensor and the attention map is supposed to tell me where in the image does the model think have the best exposure. I wonder if there is a way to visualize this attention, looking like this:
image
Below are my image and its attention map.


Thanks for your advice.

1 Like

Does anyone have a clue pls?

1 Like