1. RKNN Model Zoo 是什么?
RKNN Model Zoo 是瑞莎(Radxa)为 Rockchip NPU(RKNPU)开发者提供的模型资源库和工具集合。它基于 RKNPU SDK 工具链开发,包含了当前主流 AI 算法的部署示例,覆盖了从模型转换到推理部署的完整流程。
这个工具集主要解决了两大痛点:
- 为 Rockchip 芯片开发者提供开箱即用的模型部署方案
- 降低了从训练框架到嵌入式设备的部署门槛
在实际项目中,我发现很多开发者最头疼的不是模型训练,而是如何将训练好的模型高效部署到嵌入式设备上。RKNN Model Zoo 通过提供标准化的流程,大大简化了这个过程。
2. 环境准备与工具链配置
2.1 硬件需求
要使用 RKNN Model Zoo,你需要准备:
- 开发主机:x86 Linux PC(推荐 Ubuntu 18.04/20.04)
- 目标设备:搭载 Rockchip NPU 的开发板(如 RK3566、RK3588等)
- 连接线缆:USB转串口调试线、网线等
注意:不同型号的 Rockchip 芯片对应的 NPU 算力差异较大,在选择开发板时要根据模型的计算量需求合理选择。
2.2 软件依赖安装
首先需要搭建基础开发环境:
bash复制# 安装基础工具
sudo apt update
sudo apt install -y git cmake python3 python3-pip python3-venv
# 创建工作目录
mkdir RKSDK && cd RKSDK
# 克隆仓库(指定版本)
git clone -b v2.3.2 https://github.com/airockchip/rknn_model_zoo.git
然后配置 Python 虚拟环境:
bash复制# 创建虚拟环境
python3 -m venv rknn_env
source rknn_env/bin/activate
# 安装 RKNN-Toolkit2
pip install rknn-toolkit2
2.3 交叉编译工具链配置
对于需要在 x86 主机上编译 ARM 目标程序的情况,需要配置交叉编译工具链:
bash复制#
