AI挑战营（进阶）onnx转rknn

sgf201

AI挑战营（进阶）onnx转rknn [复制链接]

一、安装rknn-toolkit

下载rknn-toolkit2.git

git clone

链接已隐藏，如需查看请登录或者注册

安装依赖项目

cd packages/x86_64/

pip install -r requirements_cp310-2.3.0.txt 

报错如下：

 

看了下安装的pytorch发现版本对不上，我torch装的是2.4.0

回退rknntoolkit的版本到2.2.0

pip install -r requirements_cp310-2.2.0.txt  

执行成功

pip install rknn_toolkit2-2.2.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl 

二、下载rknn_model_zoo

git clone

链接已隐藏，如需查看请登录或者注册

cd examples/yolov5/model

 

bash ./download_model.sh 

python convert.py ../model/yolov5s_relu.onnx rk1106 i8 ../model/yolov5s_relu.rknn 命令报下面的错误

  1106应该用rv

python convert.py ../model/yolov5s_relu.onnx rv1106 i8 ../model/yolov5s_relu.rknn

三、更新example中的rknn模型

 

替换模型后编译完成scp到小板子上运行如下：

 

 

 

Jacktang

为什么还要更新这个example中的rknn模型呢

sgf201

Jacktang 发表于 2024-12-25 07:30 为什么还要更新这个example中的rknn模型呢

我是确认下我转出来的rknn是否可用

freebsder

谢谢分享，AI模型和参数的本地化落地是个很重要的环节。