I am working on one network, for that the input is image and heatmap corresponding to 21 landmarks for face alignment. I want to preprocess the data. Could anyone suggest any method of how to generate heatmap from the given x,y coordinate in mat file in PyTorch.