def dispImageFromModel(tensor):
    img = tensor.cpu().detach().numpy()
    img = img.reshape((512,512))
    img, cb, cr = Image.fromarray(img, 'YCbCr').split()
    return img

cleanImage#01200×900 112 KB
noisyImage#01200×900 256 KB