如何将AVFrame转换为类

AVFrame是FFmpeg中用于表示音视频帧的数据结构。它包含了音视频帧的各种属性和数据，如像素格式、图像宽高、音频采样率等。

将AVFrame转换为类似于OpenCV中的Mat对象，可以通过以下步骤实现：

首先，创建一个空的Mat对象，用于存储转换后的图像数据。

cv::Mat mat;

然后，根据AVFrame的像素格式和图像宽高，确定Mat对象的像素格式和尺寸。

int width = avFrame->width;
int height = avFrame->height;
AVPixelFormat pixelFormat = avFrame->format;

int cvPixelFormat;
switch (pixelFormat) {
    case AV_PIX_FMT_BGR24:
        cvPixelFormat = CV_8UC3;
        break;
    case AV_PIX_FMT_GRAY8:
        cvPixelFormat = CV_8UC1;
        break;
    // 其他像素格式的处理
    default:
        // 不支持的像素格式
        return;
}

mat.create(height, width, cvPixelFormat);

接下来，将AVFrame的数据复制到Mat对象中。

uint8_t* data = avFrame->data[0];
int linesize = avFrame->linesize[0];

for (int i = 0; i < height; i++) {
    memcpy(mat.ptr(i), data + i * linesize, width * mat.elemSize());
}