5.5.2 模型推理
模型推理开发
功能背景
hobot_dnn
是TogetheROS.Bot软件栈中的板端算法推理框架,在RDK上利用BPU处理器实现算法推理功能,基于D-Robotics算法推理框架和ROS2 Node进行二次开发,为机器人应用开发提供更简单易用的模型集成开发接口,包括模型管理、基于模型描述的输入处理及结果解析,以及模型输出内存分配管理等功能。
通过阅读本章节,用户可以使用D-Robotics提供的模型,在RDK上基于hobot_dnn
创建并运行一个人体检测的算法Node。借助tros.b提供的组件,订阅摄像头采集&发布的图像,对图像进行算法推理检测出人体框后,使用多目标跟踪(multi-target tracking
,即MOT
)算法对检测框进行跟踪和目标编号分配,最终实现在PC端的Web浏览器上实时渲染展示图像、人体框检测和目标跟踪结果。
前置条件
1 RDK开发板,并且已安装好相关软件,包括:
-
Ubuntu 20.04/Ubuntu 22.04系统镜像。
-
tros.b软件包。
-
ROS2软件包构建、编译等工具。安装命令:
sudo apt install ros-dev-tools
2 RDK已安装F37或者GC4663摄像头。
3 可以通过网络访问RDK的PC。
关于hobot_dnn)的详细使用说明可以参考
hobot_dnn)代码中的README.md和接口说明文档。hobot_dnn的使用逻辑流程如下:
在不了解hobot_dnn)使用流程的情况下,用户也可以按照本章节流程使用
hobot_dnn)开发出模型推理示例。
本章节以下内容使用tros.b Foxy版本举例说明,如果您使用的是tros.b Humble版本,只需将source /opt/tros/setup.bash
命令替换为source /opt/tros/humble/setup.bash
。