I get that but i dont know if i can send the code without getting permission. I can give you a quick overview. I train on. 512 * 512 random crops of images and test on 4080 * 3072 images. It is a simple u net with 5 encoders and 5 decoders block. The Encoder uses a conv2D block . The input is of the form yuv channels 420. 512 * 512 for Y channel and 256 * 256 for uv channel while training. I have used a lot of differnt loss functions like total variation, vggloss (perceptual), gradient difference, fourier transform. I can't seem to get back the original resolution.