Thanks for this information.
Since you have the landmarks, how would you like to create the heatmaps?
Do you want to plot these landmarks as single pixels into a single channel image and concatenate it with the input or are you thinking about another approach?