描述
图像生成任务的平均ssim,psnr,lpips的计算。pred文件夹有生成的若干图片,gt文件夹真实的若干图片。要求图片名字有相关性。
要求
1.名称对应。名称一样或按照顺序。
2.图片像素MAX值为255。如果是灰度图要调整代码中的255为1.0
3.包版本分别为:
scikit-image 0.18.3
lpips 0.1.4
安装命令:
pip install scikit-image==0.18.3
以及
pip install lpips==0.1.4
核心代码
ssim
ssim_v = ssim(img1_np,img2_np, data_range=255.0, multichannel=True)
psnr
psnr_v = psnr(img1_np,img2_np, data_range=255.0)
lpips
lpips_val = lpips_model.forward(transform(img1_np).cuda(), transform(img2_np).cuda(