How to get heatmap?

Hi,
In many papers, I will see the heat map, how to get it, what is the API of pytorch?


thanks

You can convert the torch to numpy.
Use plt.imshow(arr,map=‘jet’)

This is how people find heatmap and report in paper.

plt.imshow(arr, map=‘jet’)
arr: is output of conv2d ?
thanks

1 Like

Please, tell me a bit in detail about “arr”?.

I use deeplabv3+ → predict output, so arr is output, is it right?