import cv2
import numpy as np
import math
#计算峰值信噪比
def psnr(img1, img2):
mse = np.mean( (img1/255. - img2/255.) ** 2 )
if mse < 1.0e-10:
return 100
PIXEL_MAX = 1
return 20 * math.log10(PIXEL_MAX / math.sqrt(mse))
ori_img = cv2.imread('tiger.png') #原始图片
den_img= cv2.imread('Gaussian_Blur.png') #去噪后的图片
print(psnr(ori_img,den_img))
其中ori_img = cv2.imread('tiger.png')
中的tiger.png
更改为原始图片名;den_img= cv2.imread('Gaussian_Blur.png')
中的Gaussian_Blur.png
为添加模糊后的图片。