什么是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上的文章。

相关推荐
山西正方元5 小时前
西安商家公私域联动落地:品牌私域架构拆解与本地化适配
大数据·人工智能·#西安本地运营
腾视科技-AI5 小时前
腾视科技大模型一体机解决方案:低成本私有化落地,重塑行业智能应用新格局
大数据·人工智能·科技·ai·ai大模型·腾视科技·ai算力盒
Thomas.Sir5 小时前
第26课:TensorFlow|循环神经网络RNN原理【时序数据处理、序列依赖关系讲解】
人工智能·rnn·tensorflow
Wang's Blog5 小时前
Vibe Coding一人即团队系列54:云服务器 Node.js 与 MySQL 9 环境搭建及配置指南
服务器·人工智能·mysql·node.js
测试开发Kevin5 小时前
DeepEval + Eval‑Harness 完整讲解(结合 Playwright UI 自动化例子)
人工智能·ai·langchain
张欣-男5 小时前
5分钟理解线性代数v2
人工智能·线性代数·机器学习
pnoker5 小时前
从工业软件到 AI 智能体:工业 AIoT 技术路线的系统梳理
java·人工智能·物联网·microsoft·开源·工业互联网
2501_931819705 小时前
车载语音青岛方言数据集搭建阶段 信实翻译垂直场景标准赋能青岛方言标注公司作业
人工智能·语音识别
jay神5 小时前
深度学习如何确定Batch Size的大小?
人工智能·深度学习·计算机视觉·毕业设计·课程设计·batch
Geeys5 小时前
拼多多商家客户端和网页版有哪些功能上的差异?
大数据·人工智能