全球最实用的IT互联网信息网站!

AI人工智能P2P分享&下载搜索网页发布信息网站地图

当前位置:诺佳网 > 电子/半导体 > 嵌入式技术 >

迅为RK3576开发板NPU环境搭建和使用rknn-toolkit2功能

时间:2025-06-19 10:53

人气:

作者:admin

标签: 开发板  迅为电子  rk3576 

导读:迅为RK3576开发板NPU环境搭建和使用rknn-toolkit2功能演示模型转换...

开发板采用核心板+底板结构,在我们的资料里提供了底板的原理图工程以及PCB工程,可以实现真正意义上的裁剪、定制属于自己的产品,满足更多应用场合。

迅为针对RK3576开发板整理出了相应的开发流程以及开发中需要用到的资料,并进行详细的说明。快速定位需要使用的文档,大大提升研发速度。

迅为iTOP-3576开发板采用瑞芯微RK3576高性能、低功耗的应用处理芯片,集成了4个Cortex-A72和4个Cortex-A53核心,以及独立的NEON协处理器。它适用于ARM PC、边缘计算、个人移动互联网设备及其他多媒体产品。

wKgZO2hRAN6AMkuwAAY51MXrvA4922.png

2 rknn-toolkit2 功能演示

迅为有录制rknn_toolkit2视频教程,详细的讲解了rknn_toolkit2各个API的使用,具体超 链接如下所示:

04_RKNN模型构建

05_RKNN模型评估-推理测试

06_RKNN模型评估-量化精度分析

07_RKNN模型评估-性能评估和内存评估

视频中使用的在rknn_toolkit2版本为1.4,而现在最新版本为2.3,RK会不断的对rknn_to olkit2版本进行更新,但对于api的使用方法是不会变的,所以大家要想了解rknn_toolkit2各 个API的使用方法,可以直接看上面的视频链接,在本手册中只会对rknn_toolkit2的例程进行 演示。

2.2.1 模型转换

上面也提到了rknn_toolkit2最重要的功能就是进行模型转换,将常用的Caffe、TensorFlo w、TensorFlow Lite、ONNX、DarkNet、PyTorch等模型转为RK支持RKNN模型,在rknn-t oolkit2/rknn-toolkit2/examples目录下提供了上述各个模型的转换和推理例程,如下图所示:

wKgZPGhTe2-ASKBQAACIxVfYLoc866.png

这里以pytorch的resnet18例程进行演示,首先来到knn-toolkit2/rknn-toolkit2/examples/py torch/resnet18目录下,如下图所示:

wKgZPGhTe2-AGuvPAACxgV09HnM618.png

由于默认没提供resnet18模型,所以需要手动拷贝到该目录下,具体模型存放路径为“i TOP-3576开发板\02_【iTOP-RK3576开发板】开发资料\05_NPU例程测试配套资料\05_resnet1 8模型”,拷贝完成如下图所示:

wKgZO2hTe2-AVDcwAAESy1oR2EM554.png

然后修改test.py内容,将第53行config接口的target_platform参数设置从rk3566修改为rk3576,target_platform参数决定了要构建具体SOC类型的RKNN模型,其他地方不需要改动,修改完成如下图所示:

wKgZO2hTe2-AIxFGAACuFdK5CKc518.png

然后保存退出,运行test.py,需要注意的是必须确保已经安装了rknn-toolkit2,使用上一小节描述的三种方法均可,这里使用的是第一种方法安装的rknn-toolkit2,运行过程如下所示:

wKgZO2hTe3GAHVC0AAIgdlPn9S4883.png

运行完成只会会生成RK3576对应的RKNN模型以及推理结果如下图所示:

wKgZPGhTe3GAFsVlAAIb_a7acmk670.png

wKgZPGhTe2-AA-kBAAFVu5232cs024.png

推理结果为space shuttle,推理的图片为航天飞机,如下图所示:

wKgZPGhTe3GAFwk3AAMFMPA4Rtg931.png

推理结果正确,至此对于rknn-toolkit2模型转换和推理就演示完成了,需要注意的是这里是在X86 Linux系统上通过模拟器进行推理的,并不代表在开发板的运行情况,更常用的推理测试为下一个小节的连板推理。

温馨提示:以上内容整理于网络,仅供参考,如果对您有帮助,留下您的阅读感言吧!
相关阅读
本类排行
相关标签
本类推荐

CPU | 内存 | 硬盘 | 显卡 | 显示器 | 主板 | 电源 | 键鼠 | 网站地图

Copyright © 2025-2035 诺佳网 版权所有 备案号:赣ICP备2025066733号
本站资料均来源互联网收集整理,作品版权归作者所有,如果侵犯了您的版权,请跟我们联系。

关注微信