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

相关推荐
小易测试笔记14 分钟前
芯片ESD测试仪是什么?主要有哪几种不同类型?
人工智能
counting money22 分钟前
Spring AI Alibaba 从入门到实战:构建企业级 AI 应用
java·人工智能·spring
科技每日热闻22 分钟前
AI 赋能财务资金管理、风险预警、经营数据分析,合适的云上财务智能 Agent 有哪些?—— 优先采用 Amazon Quick 四链路解决方案
人工智能·ai·数据挖掘·数据分析
Linguwen26 分钟前
外贸GEO06|AI搜索引擎怎么工作?理解原理才能做好优化
人工智能·搜索引擎
DS随心转APP29 分钟前
生成word文档的DeepSeek底层解码与“AI导出鸭”工程化方案:一份技术架构师的全维度测评
人工智能·ai·chatgpt·word·deepseek·ai导出鸭
小猴子下山12334 分钟前
2026年无锡滨湖区健康管理公司解析:全周期服务与细胞技术如何匹配需求
人工智能·精选
武子康36 分钟前
Pi 怎样决定模型看见什么:AGENTS.md、SYSTEM.md 与 Skills 的加载边界
人工智能·llm·agent
深念Y39 分钟前
Opencode Event 表写入优化方案
数据库·人工智能·ai·node.js·bug·优化·opencode
橘和柠40 分钟前
YOLOv8 目标检测:从训练到部署全流程(接单高频需求)
人工智能
招风的黑耳42 分钟前
我国牵头制定智慧城市国际标准发布:从“跟跑“到“领跑“,意味着什么?
人工智能·智慧城市