python批量计算psnr和ssim

你可以使用Python中的OpenCV库来计算图像的PSNR和SSIM值。以下是一个简单的示例代码：

import cv2
import numpy as np

# 读取图像
img1 = cv2.imread('image1.jpg')
img2 = cv2.imread('image2.jpg')

# 将图像转换为灰度图像
gray1 = cv2.cvtColor(img1, cv2.COLOR_BGR2GRAY)
gray2 = cv2.cvtColor(img2, cv2.COLOR_BGR2GRAY)

# 计算PSNR值
psnr = cv2.PSNR(gray1, gray2)
print("PSNR value is {}".format(psnr))

# 计算SSIM值
ssim = cv2.SSIM(gray1, gray2)
print("SSIM value is {}".format(ssim))

如果你需要批量计算多个图像的PSNR和SSIM值，你可以使用循环来遍历图像文件夹中的所有图像文件。以下是一个示例代码：

import cv2
import numpy as np
import os

# 图像文件夹路径
img_folder = 'image_folder/'

# 遍历图像文件夹中的所有图像文件
for filename in os.listdir(img_folder):
    # 读取图像
    img = cv2.imread(os.path.join(img_folder, filename))
    
    # 将图像转换为灰度图像
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

    # 计算PSNR值
    psnr = cv2.PSNR(gray1, gray2)
    print("PSNR value of {} is {}".format(filename, psnr))

    # 计算SSIM值
    ssim = cv2.SSIM(gray1, gray2)
    print("SSIM value of {} is {}".format(filename, ssim))

你需要将上面的代码中的`image_folder/`替换为你的图像文件夹的路径。