1.计算PSNR,SSIM,MSE
当计算出现SSIM报错,情况1
解决办法:
1.设置 multichannel=True即可。
2.按照单张图去计算SSIM,此时注意输进去的图片去掉batch这个维度,应为(H,W,C)。
3.取batch时确保每一个batch数都大于win_size(默认为7)。
4.更改win_size大小,此做法并不推荐,因为要是和别的文章中数值进行比较,win_size大小不一样应该会对结果有一定影响,还是默认大小比较好,不过我没做实验,不知道此影响多大。
运行成功!
当计算出现SSIM报错,情况2
一般出现在超分辨率之后,在不同的采样比例下,图片的尺寸发生变化,导致报错。
修改后:
成功解决!