Hi,
I would like to train a neural network for landmarks detection as heatmaps using U-Net architecture.
I have 20 points, and I have generated a binary mask for each point (20 classes + background = 21 classes).
For example, my input is 800x800x1 and I would like to have my output as 800x800x21.
Does anybody know how to deals with it?