用Python计算两图像的峰值信噪比PSNR

图像信噪比是用于评估图像超分结果的指标

  

   

    

     

    
    

     

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

1.首先计算mse。

对于三通道的RGB图像 计算mse的数学表达式是：

转换成代码来写：

2.得到mse后计算PSNR。

转换为代码来写：

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/109805423