峰值信噪比python代码
时间: 2023-07-09 22:46:32 浏览: 216
用python实现求不同分辨率图像的峰值信噪比
计算峰值信噪比(Peak Signal-to-Noise Ratio,PSNR)的Python代码如下:
```python
import cv2
import numpy as np
def psnr(img1, img2):
mse = np.mean((img1 - img2) ** 2)
if mse == 0:
return float('inf')
max_pixel = 255.0
psnr = 20 * np.log10(max_pixel / np.sqrt(mse))
return psnr
# 示例:
img1 = cv2.imread('image1.jpg')
img2 = cv2.imread('image2.jpg')
print("PSNR: ", psnr(img1, img2))
```
其中,`img1`和`img2`分别是待比较的两幅图像,可以用OpenCV库读入。函数中计算了两幅图像的均方误差(Mean Square Error,MSE),然后根据MSE计算PSNR值。如果MSE为0,说明两幅图像完全一致,此时PSNR为正无穷。否则,PSNR值越高,表示两幅图像越接近。
阅读全文