PSNR的python代码

冷不防 2022-05-09 14:20 218阅读 0赞

此文转自：[https://blog.csdn.net/breezero/article/details/82285188][https_blog.csdn.net_breezero_article_details_82285188] 侵权删。

flatten()可以将二维的array展成一维的

a.flatten(order=’C’)  
Return a copy of the array collapsed into one dimension.  
`order= 'C'` means to flatten in row-major (C-style) order.  
`order='F'` means to flatten in column-major (Fortran-style) order.

flatten('C')和flatten('F')的区别在于行向量和列向量。

图像的超分辨率重建中，计算PSNR的时候，需要将矩阵形式的二维数据变成向量形式的一维向量，这样方便计算。

# PSNR.py
     
    import numpy as np
    import math
     
     
    def psnr(target, ref, scale):
        # target:目标图像  ref:参考图像  scale:尺寸大小
        # assume RGB image
        target_data = np.array(target)
        target_data = target_data[scale:-scale,scale:-scale]
     
        ref_data = np.array(ref)
        ref_data = ref_data[scale:-scale,scale:-scale]
     
        diff = ref_data - target_data
        diff = diff.flatten('C')
        rmse = math.sqrt( np.mean(diff ** 2.) )
        return 20*math.log10(1.0/rmse)

![PSNR = 10\\cdot log\_\{10\}(\\frac\{MAX\_\{I\}^\{2\}\}\{MSE\}) = 20\\cdot log\_\{10\}(\\frac\{MAX\_\{I\}\}\{\\sqrt\{MSE\}\})][PSNR _ 10_cdot log_10_frac_MAX_I_2_MSE_ _ 20_cdot log_10_frac_MAX_I_sqrt_MSE]

[https_blog.csdn.net_breezero_article_details_82285188]: https://blog.csdn.net/breezero/article/details/82285188
[PSNR _ 10_cdot log_10_frac_MAX_I_2_MSE_ _ 20_cdot log_10_frac_MAX_I_sqrt_MSE]: https://private.codecogs.com/gif.latex?PSNR%20%3D%2010%5Ccdot%20log_%7B10%7D%28%5Cfrac%7BMAX_%7BI%7D%5E%7B2%7D%7D%7BMSE%7D%29%20%3D%2020%5Ccdot%20log_%7B10%7D%28%5Cfrac%7BMAX_%7BI%7D%7D%7B%5Csqrt%7BMSE%7D%7D%29

发表评论取消回复

表情：

评论列表（有 0 条评论，218人围观）

还没有评论，来说两句吧...

相关阅读

相关 PSNR的计算以及对应彩色图像PSNR的计算以及BDBR与BD-PSNR和SSIM

1,PSNR PSNR is most easily defined via the [mean squared error][] (MSE). Given a nois

曾经终败给现在/ 2022年12月05日 12:20/ 0 赞/ 325 阅读

相关图像质量评估指标：MSE，PSNR，SSIM

文章目录 MSE与PSNR的计算方法 MSE与PSNR的问题 SSIM的理念与计算方法理念概念准备公式理解

谁践踏了优雅/ 2022年11月21日 03:49/ 0 赞/ 1387 阅读

相关 psnr和ssim源代码（c++版）

结合网络上的资料和自己的修改，实现了psnr和ssim的c++版本。 include<iostream> include<opencv2\opencv

╰+哭是因爲堅強的太久メ/ 2022年08月08日 05:30/ 0 赞/ 290 阅读

相关 psnr ssim图像评价指标解释

全参考客观视频质量评价方法是指把原始参考视频与失真视频在每一个对应帧中的每一个对应像素之问进行比较。准确的讲，这种方法得到的并不是真正的视频质量，而是失真视频相对于原始视频的相

r囧r小猫/ 2022年08月08日 05:28/ 0 赞/ 325 阅读

相关图像质量的客观评估指标PSNR与SSIM

PSNR SSIM 代码参考文献 -------------------- 1：PSNR PSNR是最为常用的图像质量评估指标： ![这

怼烎@/ 2022年07月28日 05:09/ 0 赞/ 725 阅读

相关计算两个YUV视频序列的PSNR

/ Calculate PSNR between 2 YUV420P file @param url1 Location of firs

╰+攻爆jí腚メ/ 2022年06月13日 06:13/ 0 赞/ 202 阅读

相关 PSNR峰值信噪比matlab实现

PSNR，峰值信噪比，通常用来评价一幅图像压缩后和原图像相比质量的好坏，当然，压缩后图像一定会比原图像质量差的，所以就用这样一个评价指标来规定标准了。PSNR越高，压缩后失真越

超、凢脫俗/ 2022年06月11日 06:48/ 0 赞/ 212 阅读

相关 PSNR的python代码

此文转自：[https://blog.csdn.net/breezero/article/details/82285188][https_blog.csdn.net_breez

冷不防/ 2022年05月09日 14:20/ 0 赞/ 219 阅读

相关计算matlab中图像的PSNR和SSIM

网上找了很多关于PSNR和SSIM的计算，很多结果算出来都不一样，公式都是普遍的，如下： ![Center][] 现在总结下造成结果差异的原因。 PSNR的差异

Dear 丶/ 2022年05月08日 11:50/ 0 赞/ 400 阅读

相关关于MAE MSE PSNR SSIM 的含义

MAE（Mean Absolute Error）平均绝对误差可以更好地反映预测值误差的实际情况。即 L1 loss ![MAE(X, h)=\\frac\{1\}\{m\

╰半夏微凉°/ 2021年12月23日 07:49/ 0 赞/ 503 阅读