OpenCV buffer转cv::Mat


void convertBuf2Mat(void* input_ptr, vx_uint32 width, vx_uint32 height, cv::Mat& frame)
{
    // 计算亮度通道和色度通道的大小
    size_t y_size = width * height;
    size_t uv_size = y_size / 2;

    // 创建一个只包含亮度通道的 cv::Mat 对象
    cv::Mat y_channel(height, width, CV_8UC1, (unsigned char*)input_ptr);

    // 设置 U 通道和 V 通道的数据指针
    unsigned char* u_ptr = (unsigned char*)input_ptr + y_size;
    unsigned char* v_ptr = u_ptr + uv_size;

    // 创建只包含 U 通道和 V 通道的 cv::Mat 对象
    cv::Mat u_channel(height / 2, width / 2, CV_8UC1, u_ptr);
    cv::Mat v_channel(height / 2, width / 2, CV_8UC1, v_ptr);

    // 调整色度通道的大小以匹配亮度通道
    cv::resize(u_channel, u_channel, cv::Size(width, height));
    cv::resize(v_channel, v_channel, cv::Size(width, height));

    // 合并亮度和色度通道,得到 NV12 格式的 cv::Mat
    std::vector channels = { y_channel, u_channel, v_channel };
    cv::merge(channels, frame);
}

你可能感兴趣的:(opencv,人工智能,计算机视觉,c++)