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

相关推荐
用户2991422196803 分钟前
GitHub 热榜项目:周榜(2026-11-08)
人工智能·开源·github
天天被压力9 分钟前
【A股分钟级高频数据完全手册 #07】北交所历史分钟K线接口:京市Pro版分时批量
java·人工智能·python
用户29914221968012 分钟前
AI 日报(2026年11月8日)
人工智能
天天被压力15 分钟前
【A股分钟级高频数据完全手册 #08】股票日、分时数据接口对比:普通分时与Pro版差异实测
java·人工智能·python
miofly15 分钟前
NVIDIA 开源 Cosmos 3 四款模型,用于交互式世界模拟与自动驾驶场景生成
人工智能
用户29914221968018 分钟前
GitHub 热榜项目:日榜(2026-11-08)
人工智能·开源·github
Dawson Zhu1 小时前
【LLM推理与Serving⑤】Prefix Cache前缀缓存:让重复prompt不再重复计算
人工智能·语言模型·架构·aigc·agi
Rocky Ding*1 小时前
深入浅出IDM-VTON论文精读:虚拟试衣怎样同时保住服装语义与局部细节
论文阅读·人工智能·深度学习·机器学习·aigc·ai-native·idm-vton
严肃小白兔1 小时前
MATLAB 数值计算中的精度与误差问题:浮点运算、误差传播与数值稳定性
人工智能·算法·matlab
AI导出鸭PC端1 小时前
Perplexity批量导出没有PDF模式?AI导出鸭插件补上最后一块拼图
人工智能·pdf·c#·ai导出鸭