opencv c++计算视频psnr

要计算视频的PSNR，需要先使用OpenCV打开视频文件并逐帧读取视频帧，然后将每一帧与参考帧进行比较计算PSNR值。下面是一个简单的C++代码示例：

#include <opencv2/opencv.hpp>
#include <iostream>

using namespace cv;
using namespace std;

double getPSNR(const Mat& I1, const Mat& I2)
{
    Mat s1;
    absdiff(I1, I2, s1); //计算两帧图像的差异
    s1.convertTo(s1, CV_32F);
    s1 = s1.mul(s1);
    Scalar s = sum(s1);
    double mse = s.val[0] / (double)(I1.size().height * I1.size().width);
    if (mse < 10e-10) return 100; //如果PSNR极高，返回100
    double psnr = 10.0 * log10((255 * 255) / mse);
    return psnr;
}

int main()
{
    VideoCapture cap("test.mp4"); //打开视频文件
    if (!cap.isOpened())
    {
        cout << "Can't open video file!" << endl;
        return -1;
    }

    Mat ref_frame, cur_frame;
    cap.read(ref_frame); //读取参考帧
    double psnr_sum = 0;
    int count = 0;

    while (cap.read(cur_frame)) //循环读取视频帧
    {
        double psnr = getPSNR(ref_frame, cur_frame); //计算PSNR
        psnr_sum += psnr;
        count++;

        //显示计算结果
        cout << "Frame " << count << ", PSNR = " << psnr << " dB" << endl;

        imshow("Current Frame", cur_frame);
        waitKey(10);

        ref_frame = cur_frame.clone(); //将当前帧保存为参考帧
    }

    double avg_psnr = psnr_sum / count;
    cout << "Average PSNR = " << avg_psnr << " dB" << endl;

    return 0;
}

在这个示例中，我们使用`VideoCapture`类打开名为`test.mp4`的视频文件，并逐帧读取视频帧。在每一帧中，我们使用`getPSNR`函数计算当前帧和参考帧之间的PSNR值。最后，我们计算所有帧的平均PSNR值，并将结果输出到控制台。