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推理模型转换及测试过程,希望能够对您有所帮助。进入飞凌嵌入式官网,即可了解有关RK3588-C开发板的更多产品详情。