Hi
This is somehow a simple UNet implementation, you can use available great UNet tutorials like This to code your model in Pytorch.
The main difference might be your input and output image where you should add noise to have those noisy inputs.
do you have any template code implementation regarding the same? I am taking ultrasound nerve segmentation dataset for speckle noise removal from ultrasound images using batch renormalization.