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

2025/2/7 3:52:45 来源：https://blog.csdn.net/jcfszxc/article/details/142212206 浏览: 次关键词：【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_rectim_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 格式// 确保矩形尺寸至少为 2x2if (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硬件加速在给定的图像数据上绘制一个矩形框。它主要用于在图像处理或计算机视觉应用中标记目标区域。

函数步骤：

创建并初始化 rga_buffer_t 结构体，用于描述目标图像缓冲区。
使用 wrapbuffer_virtualaddr 函数设置虚拟地址。
将OpenCV的 cv::Rect 转换为RGA使用的 im_rect 结构。
设置矩形框的颜色（本例中为绿色）。
确保矩形尺寸至少为2x2像素。
使用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;
}