good job, is it possible to add other datasets for training and testing and also is there any sample of PSNR and SSIM evaluation metrics, thanks!