Nevermind, I see the “heatmaps” reference now. So you’re wanting the NN to replicate the heatmap from something like an MRI. Correct? Why are the number of features out equal to 17?
That seems a little odd. A heatmap should have just one channel.