海思35XX-YUV420转RGB888

一、函数功能和作用

ive_yuv2rgb 函数的主要功能是将输入的 YUV 格式的视频帧转换为 RGB 格式的 cv::Mat 对象，以便在后续的图像处理和分析中能够更方便地使用。这种颜色空间的转换在许多应用中都非常重要，例如视频处理、图像增强、目标检测等。YUV 是一种常见的视频压缩格式，而 RGB 则更适合于大多数图像处理算法和显示设备。

二、实现步骤

1. 初始化输入图像参数

   • 首先，定义了一个 ot_svp_src_img 类型的变量 srcImage 来表示输入图像。
   • 通过 memset 函数将其内存初始化为 0 。
   • 然后，从输入的视频帧信息 frame 中获取宽度、高度、图像类型、物理地址和步长等参数，并设置给 srcImage 。

2. 初始化输出图像参数

   • 类似地，定义并初始化了 ot_svp_src_img 类型的变量 dstImage 作为输出图像。
   • 使用 sample_common_ive_create_image_flag 函数为输出图像创建相关的标志和参数。

3. 初始化颜色空间转换控制参数

   • 定义了 ot_ive_csc_ctrl 类型的变量 cscCtrl 。
   • 将其模式设置为 OT_IVE_CSC_MODE_PIC_BT601_YUV_TO_RGB ，指定了从 YUV 到 RGB 的转换模式。

4. 创建 IVE 任务句柄并执行转换

   • 创建了 ot_ive_handle 类型的变量 iveHandle 。
   • 使用 ss_mpi_ive_csc 函数执行颜色空间转换，并通过返回值判断转换是否成功。如果不成功，会打印错误代码。

5. 数据处理与返回

   • 通过 sample_svp_convert_addr_to_ptr 函数获取输出图像的临时指针 ptr_tmp 。
   • 创建一个与输出图像尺寸和类型匹配的 cv::Mat 对象 mat 。
   • 使用 memcpy 函数将转换后的图像数据复制到 mat 中。
   • 最后，释放输出图像的物理地址，并返回转换后的 mat 对象。

总的来说，ive_yuv2rgb 函数通过一系列的参数设置和调用特定的函数，实现了高效准确的 YUV 到 RGB 的颜色空间转换，为图像处理流程提供了关键的支持。

/**
 * @brief 将 YUV 格式的视频帧转换为 RGB 格式的 OpenCV 矩阵
 * 
 * @param frame 指向视频帧信息的指针
 * @return cv::Mat 转换后的 RGB 格式的 OpenCV 矩阵
 */
cv::Mat ive_yuv2rgb(ot_video_frame_info* frame)
{
    // 初始化输入图像的参数
    ot_svp_src_img srcImage;
    memset(&srcImage, 0, sizeof(ot_svp_src_img)); // 初始化内存为 0
    srcImage.width = frame->video_frame.width; // 设置输入图像的宽度
    srcImage.height = frame->video_frame.height; // 设置输入图像的高度
    srcImage.type = OT_SVP_IMG_TYPE_YUV420SP; // 设置输入图像的类型为 YUV420SP
    srcImage.phys_addr[0] = frame->video_frame.phys_addr[0]; // 设置输入图像的物理地址
    srcImage.phys_addr[1] = frame->video_frame.phys_addr[1]; 
    srcImage.stride[0] = frame->video_frame.stride[0]; // 设置输入图像的步长
    srcImage.stride[1] = frame->video_frame.stride[1]; 

    // 初始化输出图像的参数
    ot_svp_src_img dstImage;
    memset(&dstImage, 0, sizeof(ot_svp_src_img)); 
    sample_common_ive_create_image_flag(&dstImage, OT_SVP_IMG_TYPE_U8C3_PACKAGE, frame->video_frame.width, frame->video_frame.height, TD_FALSE);

    // 初始化颜色空间转换控制参数
    ot_ive_csc_ctrl cscCtrl;
    memset(&cscCtrl, 0, sizeof(ot_ive_csc_ctrl));
    cscCtrl.mode = OT_IVE_CSC_MODE_PIC_BT601_YUV_TO_RGB; // 设置颜色空间转换模式

    // 创建 IVE 任务句柄
    ot_ive_handle iveHandle;
    int ret = ss_mpi_ive_csc(&iveHandle, &srcImage, &dstImage, &cscCtrl, TD_TRUE);
    if (ret!= 0)
    {
        printf("ss_mpi_ive_csc failed with error code: %x\n", ret); // 如果转换失败，打印错误代码
    }

    td_u8* ptr_tmp = sample_svp_convert_addr_to_ptr(td_u8, dstImage.virt_addr[0]);
    cv::Mat mat(frame->video_frame.height, frame->video_frame.width, CV_8uC3);
    memcpy((td_u8*)mat.data, (td_u8*)ptr_tmp, frame->video_frame.width * frame->video_frame.height * 3); // 复制数据到 OpenCV 矩阵
    ss_mpi_sys_mmz_free(dstImage.phys_addr[0], dstImage.virt_addr);
    return mat;
}