赞
踩
目录
开发板环境
开发环境 | 软件版本/配置 |
---|---|
开发板 | firefly rk3588J |
操作系统 | openEuler 20.03 LTS |
python版本 | 3.9.18 |
虚拟机环境
开发环境 | 软件版本/配置 |
---|---|
操作系统 | Ubuntu 18.04.6 LTS (查看命令:lsb_release -a) |
python版本 | 3.6 |
rknn-toolkit版本 | rknn-toolkit2-1.4.0 |
思路:在虚拟机上将pt模型转化成onnx模型,然后通过rknn-toolkit将onnx模型转换成rknn模型
下载 rknn-toolkit2-1.4.0.zip,yolov5-v5.0
- # 创建python环境
- conda create -n rk3588 python=3.6
- conda activate rk3588
-
- pip install numpy==1.19.5
- git clone https://gitcode.net/mirrors/rockchip-linux/rknn-toolkit2.git (下载rknn-toolkit2)
- cd ../rknn-toolkit2/doc
- pip install -r rknn-toolkit2/doc/requirements_cp36-1.4.0.txt
- pip install pandas pyyaml matplotlib seaborn
-
- # 安装rknn-toolkit2工具包
- cd ../rknn-toolkit2/packages
- pip install rknn_toolkit2-1.4.0_22dcfef4-cp36-cp36m-linux_x86_64.whl
python环境验证是否安装成功
from rknn.api import RKNN
下载yolov5v6.0
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。