RK3588的模型转换讲解

罗罗诺亚

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

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

​

  i" V- P9 h5 i5 p
. ]# e& \6 P/ Z' Y) g( _$ C+ Y1 ~

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

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

开发环境：Ubuntu20.04

01

下载RKNN-Toolkit2

​

02

安装依赖

requirements_cp36-1.3.0.txt文件，在rknn-toolkit2/doc目录下：

​

2 d* S5 V4 M+ P' H, s
4 c9 q" ^8 q5 h3 n( C

03

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

3 o$ O) e( {9 P  M
      
•       

  开发环境中安装adb

  
  

​

$ b; W' m0 R, ^, J0 o2 L1 w
      
•       

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

  
  

4 \$ I0 \5 `: F# b4 g' m
) m3 l* f* u/ g$ Q- R      
•       

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

  
  

​

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

​

04

下载NPU工程

​

4 C5 u$ s& |: `8 Y, B# l
9 U# k6 V* g: b8 Z) \7 r. v

05

将rknn_server和rknn库发送到开发板

​

- G' p+ L. c6 t- {

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

​

6 O* _5 t6 A% ?8 e( t* T( b) h4 c

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

​

! C! G& Y( W& N+ e

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

) a& D& o, o% U/ i' n  G+ i6 h

06

模型转换

3 l7 ^7 u- \9 l& X. B/ m: H
  N; G8 m+ V' x9 H9 c      
•       

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

  
  

      
•       

  修改test.py

  
    r& B/ c+ h2 k0 d  C, a

​

- H/ m8 m1 X( V

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

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

+ U! k4 x' Y4 I% A      
•       

  修改完成后，运行test.py

  
  

​

& h, x4 |  ~1 @2 q6 r

运行成功结果如下：

​

​

/ K( k+ W5 {0 e1 y5 B
: U6 |9 \( ~0 m; F- D* E

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

​

  G$ N" l# ]- f, }$ l& v2 U

07

编译测试源码

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

​

! O) w4 ^  g! V  T' K9 l; ~
" Q% R" b, W- \% p: q% z$ `

执行编译脚本，进行编译：

​

, Z- y$ `' |: d+ n3 Y: o+ |1 n" C/ `; D
/ E- C% C& ]* j

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

​

, m+ |3 l0 [/ t6 n

08

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

​

  X. G' x* |/ s' M9 ?3 y

进入到rknn_yolov5_demo_Linux目录下，添加链接库的环境变量（rknn_yolov5_demo_Linux目录下的lib目录）

​

) T& q2 b, h7 a1 W# ~# u# b

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

​

5 m/ M( Z$ L7 ^$ h' n6 f- F

执行结果如下：

​

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

​

1 P! N" }% x6 V

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

+ I5 y6 Z% Z, Q; D1 x, Y. n

mnfvbnk

图片看不见，请上传一下，我们公司可能会买这个板子