Hi, @RicCu thank you so much for the detailed reply!!
Yes my network is an encoder-decoder architecture and it will output pixel-wise predictions. Sorry I should’ve mentioned that at the first place.
Thanks again!