Heatmap loss function?

[test_server]

@MohamedAliRashad I want to do same. Can you share your code? It will help a lot