请教SSIM指标计算代码

htyjers / StrDiffusion

[CVPR 2024] Structure Matters: Tackling the Semantic Discrepancy in Diffusion Models for Image Inpainting

Apache License 2.0

85 stars 11 forks source link

Closed wangupupup closed 3 months ago

wangupupup commented 3 months ago

您好！我们的工作PSNR和FID和StrDiffusion指标接近，但ssim指标差的较多，因此想确认一下是否是因为我们的ssim指标计算的代码有问题。请问能分享一下SSIM指标计算的代码吗？

htyjers commented 3 months ago

我一直用的是这篇文章提供的指标计算方法，你可以参考这里面的ssim的参数设置方法

wangupupup commented 3 months ago

非常感谢您的回复和分享！！