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:

Below are my image and its attention map.

5Enhanced_Input.jpg512×512 51.8 KB


7A1.jpg512×512 3.4 KB

Thanks for your advice.