Python计算两张图像的PSNR
时间: 2023-12-01 13:04:59 浏览: 193
使用python和cv2进行图像拼接并计算PSNR
要计算两张图像的PSNR,可以使用Python中的OpenCV库。以下是一个示例代码:
```
import cv2
import numpy as np
# 读入两张图像
img1 = cv2.imread("image1.jpg")
img2 = cv2.imread("image2.jpg")
# 计算MSE
mse = np.mean((img1 - img2) ** 2)
# 计算PSNR
if mse == 0:
psnr = 100
else:
psnr = 10 * np.log10((255 ** 2) / mse)
print("PSNR: {:.2f}".format(psnr))
```
需要注意的是,两张图像的大小和分辨率应该相同,否则计算出的PSNR可能不准确。
阅读全文