Thanks for the answer @ptrblck. I opted for upsampling the output image as I looked in other examples they are using this approach. Actually i’m working on replicating a model from a paper for depth estimation.