什么是RKNN?

RKNN(Rockchip Neural Network) 是由瑞芯微电子(Rockchip)推出的一套针对其芯片(如 RK3399、RK3566/76 等嵌入式处理器)优化的神经网络推理引擎及工具链。它包含:

  • RKNN Toolkit :用于将主流深度学习框架(如 TensorFlow、PyTorch、ONNX)训练好的模型,转换成适合 Rockchip 芯片上运行的 RKNN 格式模型文件(一般扩展名是 .rknn)。
  • RKNN Runtime :集成在嵌入式设备上的推理库,用于加载 .rknn 模型文件,接受输入数据,执行推理,输出结果。

.rknn 文件是已经经过转换和优化的模型文件,包含了:

  • 模型结构(网络层的定义)
  • 训练好的权重参数

以及针对 Rockchip芯片硬件特性做的优化信息(如量化参数、算子融合等)。

这是一个二进制格式的封装 ,方便在设备上快速加载和推理,而不需要原始训练代码或框架环境。

使用 RKNN 的流程

  1. 模型转换(通常在PC端操作)
  • 先用 RKNN Toolkit 将 TensorFlow/PyTorch 等训练好的模型文件转换为 .rknn 文件。
  • 这个过程会进行模型结构转换、权重打包、量化(可选)、优化。
  1. 部署到设备
  • 将生成的 .rknn 文件拷贝到目标设备(嵌入式板卡)。
  1. 设备端推理
  • 使用 RKNN Runtime API(提供的 C/C++/Python 接口),加载 .rknn 文件。
  • 输入预处理后的数据(如图像),调用推理接口。
  • 获取输出结果(预测框、类别概率等)。

ps:后期会出一篇将Mixformerv2 从pt 转 onnx 转rknn,最后部署在rk3576上的文章。

相关推荐
Guheyunyi2 分钟前
视频安全监测系统的三大核心突破
大数据·运维·服务器·人工智能·安全·音视频
石像鬼₧魂石4 分钟前
HexStrike AI 理想操作流程清单(完整功能版)
linux·人工智能·windows·学习·ubuntu
阿里云大数据AI技术23 分钟前
【NeurIPS2025】阿里云PAI团队动态数据调度方案Skrull 入选
人工智能
硬汉嵌入式24 分钟前
VisualGDB 6.1 Beta5版本,正式引入全新的高速AI编辑引擎,专为C/C++项目量身打造
人工智能·visualgdb
乾元35 分钟前
AI 驱动的入侵检测与异常会话判别:从规则到行为分析前言:从“捕获敌人”到“守卫秩序”
运维·网络·人工智能·网络协议·安全
泰迪智能科技011 小时前
分享|深化产教融合丨图书联合编写招募直播
人工智能
沐雪架构师1 小时前
OpenAgents:让AI智能体Agent像人类一样联网协作
人工智能
我要充满正能量1 小时前
拥抱AI Coding,让我更自信能胜任我的工作
人工智能·ai编程·claude
安达发公司1 小时前
安达发|效率革命:APS自动排程,为“金属丛林”安装精准导航
大数据·运维·人工智能·aps高级排程·aps排程软件·安达发aps·aps自动排程
神州问学1 小时前
AI 智能体攻陷软件工程:从 SWE-Agent 到 SWE-Swiss,全景解析 AI4SE 最新战局
人工智能