c++ - 使用 OpenCV 的 C++ 图像的 SNR

Question

如何使用 OpenCV 在 C++ 中获取图像的信噪比？

我有原始图像和另一个带有高斯噪声的图像。

Answer 1

引自the OpenCV example here ，其中解释了如何从 MSE(均方误差)推导出 PSNR(峰值信噪比)，您可以使用以下函数:

double getPSNR(const Mat& I1, const Mat& I2)
{
    Mat s1;
    absdiff(I1, I2, s1);       // |I1 - I2|
    s1.convertTo(s1, CV_32F);  // cannot make a square on 8 bits
    s1 = s1.mul(s1);           // |I1 - I2|^2

    Scalar s = sum(s1);         // sum elements per channel

    double sse = s.val[0] + s.val[1] + s.val[2]; // sum channels

    if( sse <= 1e-10) // for small values return zero
        return 0;
    else
    {
        double  mse =sse /(double)(I1.channels() * I1.total());
        double psnr = 10.0*log10((255*255)/mse);
        return psnr;
    }
}

再次引用链接页面(供后人引用)，解释如下: