代码
import cv2 import math import numpy as np def psnr(img1, img2): mse = np.mean((img1 / 1023 - img2 / 1023) ** 2) if mse < 1.0e-10: return 100 PIXEL_MAX = 1 return 20 * math.log10(PIXEL_MAX / math.sqrt(mse)) def mpsnr(x_true, x_pred): n_bands = x_true.shape[2] p = [psnr(x_true[:, :, k], x_pred[:, :, k]) for k in range(n_bands)] return np.mean(p) img2 = cv2.imread('D:/project_hdr/hdr-expandnet/test_data_hdr/5892_256x256.exr') image_e_2 = cv2.imread('D:/project_hdr/hdr-expandnet/test_data_ldr/933/5892_256x256_prediction.hdr') e2 = (mpsnr(img2,image_e_2)) print(e2)这里我测试的是exr图片，普通图片要把1023改成255就可以了 结果截图
