I tried to apply it to the reconstructed images from my model, but it seems that the refined images have changed in the range of values, which affects the PSNR evaluation somewhat. I tried to set the lambda to 1, but the range of values refined is still close to twice the original value.