在图像处理,信息隐藏中,常用到PSNR来衡量图片质量。做作业遇到,记录一下。
def psnr2(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))
coverLena=cv2.imread("coverLena.jpg")
markResult=cv2.imread("markResult.jpg")
print (psnr2(coverLena , markResult))