Rk3588 onnx转rknn,出现 No module named ‘rknn‘
一、操作步骤:
rk3588 需要将yolo11 的模型onnx转rknn。
https://github.com/airockchip/rknn_model_zoo/tree/main/examples/yolo11
这个是用yolo11训练的模型,有80种类型。
完整下载下来后,在按文档描述下载模型下来:
然后进入到python文件夹
进行转换:
输入:
二、解决办法:
在https://github.com/airockchip/rknn-toolkit2 下载最新rknn 库。下载完成后,进入rknn-toolkit2-master目录
执行:
pip3 install --user -r rknn-toolkit2/packages/x86_64/requirements_cp312-2.3.0.txt
安装下载完成后,执行:
pip3 install --user rknn-toolkit2/packages/x86_64/rknn_toolkit2-2.3.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
三、安装完成后验证:
在终端输入python,然后引用rknn 模块:
from rknn.api import RKNN
如图:
这样就是没有问题的