利用RK3588深度学习模型转换

只看该作者 · 2023-6-5 09:17

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

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

RAEAOw==" title="点击并拖拽以移动" width="15" style="cursor: pointer;">编辑

下面，我们就使用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推理模型转换及测试过程，希望能够对您有所帮助。

利用RK3588深度学习模型转换

相关下载

相关帖子