图像峰值信噪比PSNR计算模板

最新推荐文章于 2021-10-29 17:03:55 发布

haohulala

最新推荐文章于 2021-10-29 17:03:55 发布

阅读量423

点赞数 1

分类专栏：计算机视觉

本文链接：https://blog.youkuaiyun.com/haohulala/article/details/112380610

版权

计算机视觉专栏收录该内容

36 篇文章

订阅专栏

本文介绍了两种计算图像峰值信噪比（PSNR）的方法，包括基本公式和修改版，通过PIL库操作图像并计算img1与img2之间的PSNR，适用于图像处理质量评估。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

参考博客 https://blog.youkuaiyun.com/u010886794/article/details/84784453

import numpy as np
import math
from PIL import Image


def psnr1(img1, img2):
    mse = np.mean((img1 - img2) ** 2)
    if mse < 1.0e-10:
        return 100
    return 10 * math.log10(255.0 ** 2 / mse)


def psnr2(img1, img2):
    mse = np.mean((img1 - img2) ** 2)
    if mse < 1.0e-10:
        return 100
    PIXEL_MAX = 255
    return 20 * math.log10(PIXEL_MAX / math.sqrt(mse))

if __name__ == "__main__":
    img1 = Image.open("img1.jpg")
    img2 = Image.open("img2.jpg")
    img1 = img1.resize((300,500))
    img2 = img2.resize((300,500))

    img1 = np.array(img1)
    img2 = np.array(img2)

    psnr1 = psnr1(img1, img2)
    psnr2 = psnr2(img1, img2)
    print(psnr1)
    print(psnr2)

img1