How to visualize the heatmap output from the hrnet?

[gjgjgjfff]

[ligaoqi2]

https://github.com/ligaoqi2/HRNet-with-Visualized-Heatmap-and-Intermediate-Results/blob/master/demo/demo_heatmap.py
You can refer to this.