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

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

convert_and_draw_rectangle 函数

功能

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

语法

cpp 复制代码
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像素。

示例用法

cpp 复制代码
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中进行适当的格式调整。

相关推荐
刘卜卜&嵌入式17 分钟前
C++_设计模式_观察者模式(Observer Pattern)
c++·观察者模式·设计模式
h汉堡41 分钟前
C++入门基础
开发语言·c++·学习
XINVRY-FPGA1 小时前
XCZU7EG‑L1FFVC1156I 赛灵思XilinxFPGA ZynqUltraScale+ MPSoC EG
c++·嵌入式硬件·阿里云·fpga开发·云计算·fpga·pcb工艺
_GR2 小时前
2025年蓝桥杯第十六届C&C++大学B组真题及代码
c语言·数据结构·c++·算法·贪心算法·蓝桥杯·动态规划
mahuifa3 小时前
(7)VTK C++开发示例 --- 使用交互器
c++·vtk·cmake·3d开发
光算科技3 小时前
服务器在国外国内用户访问慢会影响谷歌排名吗?
运维·服务器·c++
大炮筒3 小时前
CPPlist初识
数据结构·c++·list
点云SLAM5 小时前
C++中的算术转换、其他隐式类型转换和显示转换详解
c++·static_cast·dynamic_cast·c++中的类型转换·算术类型转换·其他隐式类型转换·显示类型转换
Zfox_5 小时前
Git 进阶之路:高效协作之分支管理
大数据·linux·运维·c++·git·elasticsearch
wenchm5 小时前
细说STM32单片机FreeRTOS任务管理API函数vTaskList()的使用方法
c语言·c++·stm32·单片机·嵌入式硬件