不同像素格式的两张图片生成的PSNR分数准确吗?
创始人
2025-01-09 22:00:24
0次

不同像素格式的两张图片之间不能直接计算 PSNR 分数,需要先将它们转换为相同的像素格式。下面是使用 Python Pillow 库进行像素格式转换的示例代码:

from PIL import Image

# 加载两张图片
img1 = Image.open('image1.png').convert('RGB')
img2 = Image.open('image2.jpg').convert('RGB')

# 将图片转换为相同的像素格式
if img1.mode != img2.mode:
    img2 = img2.convert(img1.mode)

# 计算 PSNR 分数
mse = calculate_mse(img1, img2)
psnr = calculate_psnr(mse)

# 计算 MSE
def calculate_mse(img1, img2):
    mse = 0
    width, height = img1.size
    for x in range(width):
        for y in range(height):
            r1, g1, b1 = img1.getpixel((x, y))
            r2, g2, b2 = img2.getpixel((x, y))
            mse += (r1 - r2) ** 2
            mse += (g1 - g2) ** 2
            mse += (b1 - b2) ** 2
    mse /= (width * height * 3)
    return mse

# 计算 PSNR 分数
def calculate_psnr(mse):
    # 计算最大可能的像素值
    max_pixel_value = 255
    # 计算 PSNR 分数
    return 20 * math.log10(max_pixel_value) - 10 * math.log10(mse)

注意:这个示例代码只适用于 RGB 格式的图片,如需处理其他格式的图片需要修改相关代码。

相关内容

热门资讯

攻略讲解辅助挂!德普之星私人局... 一款 AI 游戏攻略黑科技工具上线,查攻略效率直接开挂。它整合版本更新、装备搭配、BOSS 打法、牌...
玩家教你辅助器!wepoker... AI 黑科技赛道又上新,这款智能处理工具堪称效率开挂,专门面向短视频创作者打造。它打通脚本撰写、AI...
我来教大家开挂!wpk有作弊吗... AI 黑科技赛道又上新,这款智能处理工具堪称效率开挂,专门面向短视频创作者打造。它打通脚本撰写、AI...
热点推荐透视!wpk辅助器下载... 很多人使用云端 AI 时,都担心wpk辅助器下载、内部资料上传带来隐私风险。这款本地运行的 AI 黑...
可靠技巧透视!wepoker免... 『标题:当前标题』主打赛后复盘的 AI 游戏攻略黑科技走红,对局总结能力开挂。上传对局回放或者战绩截...
玩家交流辅助器!wpk透视辅助... 很多人使用云端 AI 时,都担心德普之星透视、内部资料上传带来隐私风险。这款本地运行的 AI 黑科技...
玩家必看分享辅助挂!wejok... 一款 AI 游戏攻略黑科技工具上线,查攻略效率直接开挂。它整合版本更新、装备搭配、BOSS 打法、牌...
我来教教你辅助挂!wepoke... 我来教教你辅助挂!wepoker辅助器有哪些功能,微扑克究竟有没有挂下载(原来是有辅助挂)AI 智能...
玩家实测辅助!wepoker有... AI 黑科技赛道又上新,这款智能处理工具堪称效率开挂,专门面向短视频创作者打造。它打通脚本撰写、AI...
科技通报透视挂!安装不了wep... AI 黑科技赛道又上新,这款智能处理工具堪称效率开挂,专门面向短视频创作者打造。它打通脚本撰写、AI...