技术帖——飞凌嵌入式RK3588开发板推理模型转换及测试

news/2025/2/12 8:18:53/

RKNN(Rockchip Neural Network)是一种用于嵌入式设备的深度学习推理框架,它提供了一个端到端的解决方案,用于将训练好的深度学习模型转换为在嵌入式设备上运行的可执行文件。使用RKNN框架可以在嵌入式设备上高效地运行深度学习模型,这对于需要在资源受限的设备上进行实时推理的应用场景非常有用。例如,可以将RKNN用于智能摄像头、机器人、无人机等嵌入式设备中,实现物体检测、人脸识别、图像分类等人工智能功能。

RKNN-Toolkit2是为用户提供在 PC、 Rockchip NPU 平台上进行模型转换、推理和性能评估的开发套件,用户通过该工具提供的 Python 接口可以便捷地完成模型转换、量化功能、模型推理、性能和内存评估以及量化精度分析等多种操作。

下面,我们就使用RKNN-Toolkit2工具将rknpu2工程中的yolov5s.onnx模型转换为yolov5s.rknn模型为例进行讲解。

开发工具:飞凌嵌入式OK3588-C开发板

开发环境:Ubuntu20.04

01:下载RKNN-Toolkit2

 

02:安装依赖

requirements_cp36-1.3.0.txt文件,在rknn-toolkit2/doc目录下:

 

03:开发环境与OK3588-C开发板连接

  • 开发环境中安装adb

  • 使用USB-typeC线连接到板子的TypeC0接口,PC端识别到虚拟机中。

  • 在开发环境中检查是否连接成功

如果连接成功会返回板子的设备ID,如下:

04:下载NPU工程

 

05:将rknn_server和rknn库发送到开发板

在OK3588-C开发板上运行rknn_server服务

在开发环境中检测rknn_server是否运行成功

有返回进程id说明运行成功。

 

06:模型转换

  • 在开发环境中进入到rknn_toolkit2工具中的examples目录中选择一个模型。本例选择的是将onnx模型转换为RKNN模型。

  • 修改test.py

  在rknn.config中添加target_platform='rk3588'

  在rknn.init_runtime中添加target='rk3588'

  • 修改完成后,运行test.py

运行成功结果如下:

同时在目录下会生成yolov5s.rknn模型。

 

07:编译测试源码

进入到rknpu2/examples/rknn_yolov5_demo目录下,设置环境变量:

执行编译脚本,进行编译:

然后在rknpu2/examples/rknn_yolov5_demo/install/rknn_yolov5_demo_Linux目录下会生成rknn_yolov5_demo

08:测试

将上边生成的yolov5s.rknn模型和install目录下的rknn_yolov5_demo_Linux拷贝到开发板中

进入到rknn_yolov5_demo_Linux目录下,添加链接库的环境变量(rknn_yolov5_demo_Linux目录下的lib目录)

使用rknn模型进行物体识别命令如下:

执行结果如下:

将生成的out.jpg拷贝到本地电脑中查看,识别结果如下。

以上就是基于飞凌嵌入式OK3588-C开发板进行的RK3588推理模型转换及测试过程,希望能够对您有所帮助。


http://www.ppmy.cn/news/162527.html

相关文章

如何实验ChatGPT 提示获取机器学习的数据集

随着机器学习的发展,获取高质量的数据集变得越来越重要。数据集对于评估最终模型的准确性和有效性至关重要,这是任何机器学习项目的先决条件。在本文中,我们将学习如何使用 ChatGPT[OpenAI] 模板提示为不同的机器学习应用程序收集各种数据集,并在 Python 中收集这些数据集。…

i3处理器_办公笔记本i3够用了吗

现在越来越多朋友会为自己办公选用一款办公笔记本,那么这款笔记本的配置是什么,办公笔记本i3够用了吗?下面就为大家带来相关的介绍。 办公笔记本i3够用了吗 1、仅是办公使用i3处理器已经足够,也就是使用的范围主要是文字、表格处理…

笔记本 续航测试软件,PCMark 10新增续航、办公测试:考验笔记本电池

UL(Futuremark)出品的PCMark 10是迄今最完整、最权威的PC综合性能基准测试工具,可以通过各种测试全方位评估一台PC的实际应用性能。 近日,官方宣布PCMark 10将会新增两个测试项目,一是万众期待的电池续航测试(Battery Life Benchmark)&#x…

2000元以内办公用计算机,2000元以内买什么笔记本 便宜实用笔记本【推荐】

2000元以内买什么笔记本? 虽然笔记本的机身十分轻便,但完全不用怀疑其应用性,在日常操作和基本商务、娱乐操作中,笔记本电脑完全可以胜任。所以笔记本也成了很多人的选择。现在又正值高考结束,想必又是一股买笔记本的热潮。下面&…

笔记本电脑快捷键

浏览器 【CtrlT】 打开一个新的标签页 【CtrlShiftT】恢复上一次关闭的标签页 【CtrlTab】同一个窗口内的多个标签页之间切换 【Alt←、→】后退和前进 【AltF4】关闭所有浏览器窗口 【CtrlF4】关闭当前浏览器窗口 【CtrlN】打开新窗口 【CtrlShiftN】打开隐私浏览模式 【Alt…

绘图用计算机配置要求,绘图笔记本电脑基本配置要求 绘图笔记本推荐

现在很多工业图、工程图都需要用电脑来话,比如CAD、PS、三维画图这些软件对电脑的配置要求还是比较高的,还有一些对色彩要求高的动画软件,那就更加需要配置了,下面小编就来简单说说绘图笔记本电脑最基本的配置要求以及绘图笔记本推…

办公计算机配置单,普通办公电脑怎么配?office三件套办公电脑主机配置推荐

对于办公用途的电脑概念比较广泛,平时做表格、码文字是办公,ps处理图片是办公、3d设计也属于办公。所以在搭配办公电脑的时候需要根据自己的具体需求去确定配置。今天电脑配置网就为大家带来一套普通办公的电脑配置单,主要用于office三件套,闲时看看电影,玩点qq棋牌类的游…

学计算机专业笔记本买多大尺寸的好,办公用笔记本电脑多大尺寸内存合适

办公用笔记本电脑多大尺寸合适 办公用本,一般对配置和性能要求不高, 如果不随身携带,建议选择屏幕大些的,否则13寸的合适。 不建议选择高分屏,办公发挥不了高分屏的优势,而且眼睛比较累。 办公用笔记本电脑…