Python计算PSNR（图像峰值信噪比）

图像信噪比是用于评估两张图像相似度的指标。

$PSNR=10\cdot \log 10\left(\frac{MA{X}_I^2}{MSE}\right)=20\cdot \log 10\left(\frac{MA{X}_I}{\sqrt{MSE}}\right)$

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))