从包含YUV420p帧的QByteArray创建一个合适的cv::Mat

#include <opencv2/opencv.hpp>

cv::Mat createMatFromYUV420p(const QByteArray& yuvData, int width, int height)
{
    int uvWidth = width / 2;
    int uvHeight = height / 2;

    cv::Mat yuvMat(height + uvHeight, width, CV_8UC1);
    cv::Mat bgrMat(height, width, CV_8UC3);

    // 提取Y分量数据
    memcpy(yuvMat.data, yuvData.constData(), width * height);

    // 提取U分量数据
    memcpy(yuvMat.data + width * height, yuvData.constData() + width * height, uvWidth * uvHeight);

    // 提取V分量数据
    memcpy(yuvMat.data + width * height + uvWidth * uvHeight, yuvData.constData() + width * height + uvWidth * uvHeight, uvWidth * uvHeight);

    // 转换为BGR格式
    cv::cvtColor(yuvMat, bgrMat, cv::COLOR_YUV2BGR_I420);

    return bgrMat;
}