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

相关推荐
丁希希哇1 分钟前
Transformer中的注意力机制(Attention)
人工智能·机器学习·transformer·大语言模型
国际学术会议-杨老师7 分钟前
2027年人文发展、城市管理与数字化社会国际会议(HUCD 2027)
大数据·人工智能·人文发展
田里的水稻15 分钟前
PyTorch_torchvision---数据集下载\TensorBoard可视化
人工智能·pytorch·python·深度学习·机器学习
科技每日热闻16 分钟前
电气设备综合维保服务商怎么选?从预防性维护到数字化运维,看懂五个关键能力——企业选择维保合作伙伴,不能只看“坏了能不能修”,更要看能否覆盖资产与系统的长期运营
大数据·运维·人工智能
Dotrust东信创智22 分钟前
HIL台架上的“无人夜班“:东信创智用AI重塑汽车软件测试全链路——测试不再是项目的瓶颈,而是交付的底气
人工智能·汽车
电子制造自留地34 分钟前
AI服务器高多层PCB设计:阻抗控制、层间对位与孔金属化工程要点
大数据·网络·人工智能·科技·pcb工艺
QYR-分析35 分钟前
汽车车窗调节器市场研究:成熟零部件赛道,模块化集成成为核心方向
人工智能·汽车
星云低代码开发平台40 分钟前
ERP 与 CRM 客户信用核对:业务键、时点和权限验收
大数据·数据库·人工智能
TF男孩42 分钟前
老AI人面对MCP、Skills等新名词
人工智能·程序员
AI Coder 码农王44 分钟前
AgentScope Java v2 第2课,ReActAgent 的循环引擎
人工智能