RKNN-Toolkit
Rockchip RKNN-ToolkitRKNN进阶瑞芯微的模型转换工具,把神经网络转成能在其 NPU 上跑的格式
RKNN-Toolkit 是瑞芯微(Rockchip)为自家芯片内置的 NPU(神经网络处理器)提供的开发工具链。它把 PyTorch、ONNX、TensorFlow 等框架训练好的模型转换成 RKNN 格式,过程中可以做 INT8 量化,再在 RK3588 等芯片上调用 NPU 推理。新芯片对应的是 RKNN-Toolkit2,板端 Python 推理用 RKNN-Toolkit-Lite2,C 接口由 RKNPU 运行时提供。很多低成本机器人和开发板用 RK3588 当主控,想在上面跑检测、分割或小型策略网络,一般都要走这套工具。
例子把训练好的 YOLO 模型导出为 ONNX,用 RKNN-Toolkit2 量化转换成 .rknn 文件,部署到 RK3588 开发板上做实时目标检测。
- 也叫
- RKNN-Toolkit2、RKNN-Toolkit-Lite2
- 相关
- 瑞芯微 RK3588、端侧部署、训练后量化、ONNX、神经网络处理器(NPU / BPU)、推理部署
- 来源
- airockchip/rknn-toolkit2 on GitHub