I managed to train a unet network and now going to do inference. Once given the test image to the network, I get a 256 channel mask which has dimension (2023,2023,256) as the output. This is related to this question.
Would anyone be able to help me in this regards
Schroter Michael