@Fahd_Jerbi have a look at U-Nets that are popular for image segmentation tasks. They make use of an ‘encoder’ half and a ‘decoder’ half that allows you to have model outputs the same size as your input.
Here’s a link you might like to go through if you want to explore it further: https://towardsdatascience.com/creating-and-training-a-u-net-model-with-pytorch-for-2d-3d-semantic-segmentation-model-building-6ab09d6a0862 2