RKNN_AI工具在RK板卡上的部署

YOLO11 在 RK3576 上部署流程(Python 3.8 推荐)

本文档整理了从环境准备到模型推理的完整流程。

备选轻量化版本工具 rknn-toolkit-lite2

192.168.3.102板卡已配置环境

1. 板卡上,环境准备

  1. Python 版本 板卡:安装python,sudo apt install *** ,推荐使用 Python 3.8 ,官方文档支持到 Python 3.12

  2. 解压并上传必要文件 将以下** 工具 ** 和 ** 示例模型 **文件上传到 RK3576板卡上:

scp ./rknn_model_zoo-2.3.0.zip develop@192.168.3.102:/home/develop/rknn_ai/ scp ./rknn-toolkit2-2.3.0.zip develop@192.168.3.102:/home/develop/rknn_ai/

工具软件https://github.com/airockchip/rknn-toolkit2.git

RKNN_AI_toolkit

示例模型https://github.com/airockchip/rknn_model_zoo/tree/v2.3.0

RKNN_AI_model_zoo

  1. 解压文件

cd ~/rknn_ai unzip rknn_model_zoo-2.3.0.zip unzip rknn-toolkit2-2.3.0.zip


2. 板卡上,安装 RKNN Toolkit 依赖

  1. 进入 ARM64 平台依赖目录:

cd ~/rknn_ai/rknn-toolkit2-2.3.0/rknn-toolkit2/packages/arm64

  1. 板卡上安装依赖:

~/rknn_ai/rknn-toolkit2-2.3.0/rknn-toolkit2/packages/arm64$: pip install -r arm64_requirements_cp38.txt

  1. 板卡上本地安装 RKNN Toolkit Wheel 包:

~/rknn_ai/rknn-toolkit2-2.3.0/rknn-toolkit2/packages/arm64$: pip install rknn_toolkit2-2.3.0-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

或明确使用本地安装使用 --no-index 禁用网络索引

pip install --no-index rknn_toolkit2-2.3.0-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

3. 板卡上,配置系统库

创建系统库目录(如果不存在):

sudo mkdir -p /usr/lib64

建立 librknnrt.so 软链接到系统库路径:

sudo ln -s ~/rknn_ai/rknn-toolkit2-2.3.0/rknpu2/runtime/Linux/librknn_api/aarch64/librknnrt.so /usr/lib64/librknnrt.so

刷新动态库缓存:

sudo ldconfig

此命令会重新扫描 /usr/lib64/ 等目录,并更新 /etc/ld.so.cache

验证是否生效

bash ldconfig -p | grep librknnrt.so

如果输出中包含 /usr/lib64/librknnrt.so,说明缓存已更新成功。

4. 准备模型文件

将模型库重命名为简短名称:

必须步骤,代码中会查找不带版本的rknn_model_zoo作为路径

mv ./rknn_model_zoo-2.3.0 ./rknn_model_zoo

下载 YOLO11 示例模型:

git clone https://github.com/airockchip/rknn_model_zoo/tree/v2.3.0
gdp@gdp:~/github/rknn_model_zoo-2.3.0/examples/yolo11/model$ : cd ~/rknn_ai/rknn_model_zoo/examples/yolo11/model chmod +x download_model.sh gdp@gdp:~/github/rknn_model_zoo-2.3.0/examples/yolo11/model$: ./download_model.sh


5. 推理运行

5.1 使用 PyTorch 或 ONNX 模型

x86主机运行

(env_isaaclab) gdp@gdp:~/github/rknn_model_zoo/examples/yolo11/model$: cd ../python (env_isaaclab) gdp@gdp:~/github/rknn_model_zoo/examples/yolo11/python$: python yolo11.py --model_path <pt_model/onnx_model> --img_show ex. python yolo11.py \ --model_path /home/gdp/github/rknn_model_zoo/examples/yolo11/model/yolo11n.onnx \ --img_show

在板卡上,yolo11n.onnx格式转换rknn_model

运行目录((base) develop@SuperShoes:~/rknn_ai/rknn_model_zoo/examples/yolo11/python)下的脚本convert.py

指令如下:

(base) develop@SuperShoes:~/rknn_ai/rknn_model_zoo/examples/yolo11/python$ python convert.py /home/develop/rknn_ai/rknn_model_zoo/examples/yolo11/model/yolo11n.onnx rk3576 fp

转换后,模型自动保存到该目录下(base) develop@SuperShoes:~/rknn_ai/rknn_model_zoo/examples/yolo11/model/yolo11n.onnx

打印脚本使用方法如下:

(base)develop@SuperShoes:~/rknn_ai/rknn_model_zoo/examples/yolo11/python$ python convert.py Usage: python3 convert.py onnx_model_path [platform] [dtype(optional)] [output_rknn_path(optional)] platform choose from [rk3562,rk3566,rk3568,rk3588,rk3576,rk1808,rv1109,rv1126] dtype choose from [i8, fp] for [rk3562,rk3566,rk3568,rk3588,rk3576] dtype choose from [u8, fp] for [rk1808,rv1109,rv1126]

5.2 使用 RKNN 模型

板卡运行 ssh @****** -Y

python yolo11.py --model_path <rknn_model> --target <TARGET_PLATFORM> --img_show ex. (base) develop@SuperShoes:~/rknn_ai/rknn_model_zoo/examples/yolo11/python$ python yolo11.py \ --model_path ~/rknn_ai/rknn_model_zoo/examples/yolo11/model/yolo11.rknn \ --target rk3576 --img_show
⚠ 注意:<rknn_model> 替换为实际 RKNN 模型路径,<TARGET_PLATFORM> 替换为目标平台,如 rk35``7``6rk3588

https://github.com/airockchip/rknn-toolkit2.git

https://github.com/airockchip/rknn_model_zoo/tree/v2.3.0

暂时无法在飞书文档外展示此内容

5.3 C++ 验证运行时库

***************************

6 参考

彬烽提供参考 【YOLO11部署至RK3588】模型训练→转换RKNN→开发板部署_yolo11 rknn-CSDN博客

主要参考官方文档 GitHub - airockchip/rknn_model_zoo at v2.3.0

GitHub - airockchip/rknn-toolkit2

相关推荐
阿里云大数据AI技术16 小时前
构建高转化海外电商搜索:阿里云OpenSearch行业算法版的全链路智能优化策略实战
人工智能·搜索引擎
Awu122717 小时前
⚡从零开发 Agent CLI(五)实现一个可治理、可扩展的工具系统
前端·人工智能·claude
字节跳动视频云技术团队17 小时前
让 Agent 成为音视频工作台:AI MediaKit CLI + Skill 发布
人工智能·音视频开发
魏祖潇17 小时前
framework 整合实战——DDD/TDD/SDD 三件套在 framework 仓的真实落地
人工智能·后端
Token炼金师17 小时前
去噪扩散:从随机噪声到高保真图像的数学之路
人工智能·aigc
这个DBA有点耶17 小时前
AI写的SQL跑崩了生产库,这锅谁背?
数据库·人工智能·程序员
阿里云大数据AI技术18 小时前
阿里云 EMR AI 助手正式发布:从问答工具到全栈智能运维助手
运维·人工智能
Larcher19 小时前
从零搭建 MCP 服务——让 AI 拥有无限扩展能力
人工智能·程序员
zzzzzz31019 小时前
你的 AI 写的 React 烂透了?这个 8000+ Star 的开源工具能揪出 90% 的「Agent 屎山」
人工智能
小星AI19 小时前
MCP协议超详细教程,从入门到实战
人工智能