I have built a model in Keras for detecting keypoints of cats. For each image, I have 3 keypoints with three corresponding heatmaps. I stacked the 3 heatmaps together to get