opencv c++计算两个视频间的psnr

要计算两个视频之间的 PSNR，您需要使用 OpenCV C++ 库。

以下是计算两个视频之间 PSNR 的简单步骤：

1. 导入 OpenCV 库

#include <opencv2/highgui.hpp>
#include <opencv2/imgproc.hpp>

2. 加载视频文件

cv::VideoCapture video1("video1.mp4");
cv::VideoCapture video2("video2.mp4");

3. 检查视频是否成功加载

if(!video1.isOpened() || !video2.isOpened()){
    std::cout << "Failed to open one or both videos" << std::endl;
    return -1;
}

4. 创建变量来存储每个视频的帧

cv::Mat frame1, frame2;

5. 创建变量来存储 PSNR 值

double psnr = 0;

6. 循环遍历两个视频的每一帧并计算 PSNR

while (true) {
    // 从第一个视频中读取一帧
    video1 >> frame1;

    // 从第二个视频中读取一帧
    video2 >> frame2;

    // 检查是否完成了视频的读取
    if (frame1.empty() || frame2.empty()) {
        break;
    }

    // 计算两帧之间的 PSNR
    psnr = cv::PSNR(frame1, frame2);
}

7. 输出 PSNR 值

std::cout << "PSNR value: " << psnr << std::endl;

完整代码示例：

#include <opencv2/highgui.hpp>
#include <opencv2/imgproc.hpp>
#include <iostream>

int main() {
    // 加载视频文件
    cv::VideoCapture video1("video1.mp4");
    cv::VideoCapture video2("video2.mp4");

    // 检查视频是否成功加载
    if (!video1.isOpened() || !video2.isOpened()) {
        std::cout << "Failed to open one or both videos" << std::endl;
        return -1;
    }

    // 创建变量来存储每个视频的帧
    cv::Mat frame1, frame2;

    // 创建变量来存储 PSNR 值
    double psnr = 0;

    // 循环遍历两个视频的每一帧并计算 PSNR
    while (true) {
        // 从第一个视频中读取一帧
        video1 >> frame1;

        // 从第二个视频中读取一帧
        video2 >> frame2;

        // 检查是否完成了视频的读取
        if (frame1.empty() || frame2.empty()) {
            break;
        }

        // 计算两帧之间的 PSNR
        psnr = cv::PSNR(frame1, frame2);
    }

    // 输出 PSNR 值
    std::cout << "PSNR value: " << psnr << std::endl;

    return 0;
}