【RKNN系列】常用函数:使用RGA加速画框

以下是针对convert_and_draw_rectangle函数的详细使用说明:

convert_and_draw_rectangle 函数

功能

在给定的图像数据上使用RGA(Rockchip Graphics Acceleration)绘制矩形框。

语法

IM_STATUS convert_and_draw_rectangle(uint8_t* dst_data, int width, int height, const cv::Rect& bbox) {
    // 创建 rga_buffer_t 结构
    rga_buffer_t dst;
    memset(&dst, 0, sizeof(rga_buffer_t));

    // 设置 rga_buffer_t 的属性
    dst.width = width;
    dst.height = height;
    dst.wstride = width;
    dst.hstride = height;
    dst.format = RK_FORMAT_BGR_888;  // 假设输入格式是 BGR

    // 使用 wrapbuffer_virtualaddr 设置虚拟地址
    dst = wrapbuffer_virtualaddr(dst_data, width, height, RK_FORMAT_BGR_888);

    // 将 cv::Rect 转换为 im_rect
    im_rect rga_rect = {
        .x = static_cast<int>(bbox.x),
        .y = static_cast<int>(bbox.y),
        .width = static_cast<int>(bbox.width),
        .height = static_cast<int>(bbox.height)
    };

    // 设置颜色 (绿色)
    uint32_t color = 0x00FF0000;  // ABGR 格式

    // 确保矩形尺寸至少为 2x2
    if (rga_rect.width < 2) rga_rect.width = 2;
    if (rga_rect.height < 2) rga_rect.height = 2;

    // 使用 RGA 绘制矩形
    IM_STATUS status = imrectangle(dst, rga_rect, color, 2);

    return status;

}

参数

  • dst_data: 目标图像数据的指针(uint8_t*)
  • width: 图像宽度(整数)
  • height: 图像高度(整数)
  • bbox: 要绘制的矩形框(cv::Rect 类型)

返回值

  • IM_STATUS: RGA操作的状态,表示操作是否成功

描述

这个函数使用RGA硬件加速在给定的图像数据上绘制一个矩形框。它主要用于在图像处理或计算机视觉应用中标记目标区域。

函数步骤:

  1. 创建并初始化 rga_buffer_t 结构体,用于描述目标图像缓冲区。
  2. 使用 wrapbuffer_virtualaddr 函数设置虚拟地址。
  3. 将OpenCV的 cv::Rect 转换为RGA使用的 im_rect 结构。
  4. 设置矩形框的颜色(本例中为绿色)。
  5. 确保矩形尺寸至少为2x2像素。
  6. 使用RGA的 imrectangle 函数绘制矩形。

注意事项

  • 函数假设输入图像格式为BGR(RK_FORMAT_BGR_888)。
  • 使用了RGA库,需要包含相应的头文件和链接库。
  • 矩形框的颜色是硬编码的绿色(ABGR格式:0x00FF0000)。
  • 函数会自动调整矩形尺寸,确保最小为2x2像素。
  • 矩形框的线宽固定为2像素。

示例用法

int main() {
    // 假设我们有一个800x600的BGR图像
    int width = 800, height = 600;
    uint8_t* image_data = new uint8_t[width * height * 3];
    
    // 填充图像数据...

    // 定义要绘制的矩形
    cv::Rect bbox(100, 100, 200, 150);

    // 调用函数绘制矩形
    IM_STATUS status = convert_and_draw_rectangle(image_data, width, height, bbox);

    if (status == IM_STATUS_SUCCESS) {
        std::cout << "Rectangle drawn successfully" << std::endl;
    } else {
        std::cerr << "Failed to draw rectangle" << std::endl;
    }

    // 清理内存
    delete[] image_data;

    return 0;
}

依赖

  • RGA库(Rockchip Graphics Acceleration)
  • OpenCV库(用于cv::Rect类型)

这个说明提供了函数的基本信息、用途、参数说明、返回值、详细描述、使用注意事项和示例用法。您可以根据需要在Obsidian中进行适当的格式调整。

你可能感兴趣的:(RKNN系列,Rockchip,rknn-toolkit2,c++,RKNN)