瑞芯微rknn-toolkit2 v1.6.0开发环境配置

拉灯是我干掉的
原创
发布时间: 2025-06-10 20:39:09 | 阅读数 0收藏数 0评论数 0
封面
本教程简单的介绍了如何在ubuntu系统中配置瑞芯微开发环境rknn-toolkit2,及其步骤的含义, 如何在x86_64 ubuntu系统中rknn-toolkit2环境下交叉编译示例工程及其部署到目标环境中。
1

按照要求建立新的conda环境

本次示例 ubuntu20.04 安装 python3.8

ubuntu 22.04 安装 python3.10

# ubuntu 20.04
conda create -n rknn-toolkit2 python=3.8
# ubuntu 22.04
conda create -n rknn-toolkit2 python=3.10
# 删除环境
conda remove -n rknn-toolkit2 --all
2

环境安装成功


3

激活环境

conda activate rknn-toolkit2


4

安装python库

由于清华源里没有这个包,单独安装

pip install tf-estimator-nightly==2.8.0.dev2021122109

再安装库

pip install -r requirements_cp38-1.6.0.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
5

安装rknn-toolkit2

pip install rknn_toolkit2-1.6.0+81f21f4d-cp38-cp38-linux_x86_64.whl


6

验证

切换目录,激活环境

conda activate rknn-toolkit2
cd ~/code/rknn/rknn-toolkit2-v1.6.0/rknn-toolkit2/examples/onnx/yolov5
python test.py
7

转到rknpu目标目录

切换目录到想要部署的模型文件夹下,本次示例为yolov5

8

下载编译工具链

apt install cmake

apt-get install gcc-aarch64-linux-gnu g++-aarch64-linux-gnu


9

使用示例脚本编译


10

复制编译安装的文件夹到linux板子

编译安装好的文件在 ./install 目录下,方式有很多,在同一个局域网络下,传输工具就可以;实在不行拿U盘拷贝也是可以的。

11

给执行权限

在linux板子上执行,之后的步骤也都是在RK3588的板子上操作。

chmod +x rknn_yolov5_demo
12

执行测试

./rknn_yolov5_demo ./model/RK3588/yolov5s-640-640.rknn ./model/bus.jpg letterbox ./result.jpg


13

测试结果


阅读记录0
点赞0
收藏0
禁止 本文未经作者允许授权,禁止转载
猜你喜欢
评论/提问(已发布 0 条)
评论 评论
收藏 收藏
分享 分享
pdf下载 下载