RKNN-Toolkit2不同版本导出的RKNN模型输出不同

最新推荐文章于 2025-06-28 09:57:16 发布
最新推荐文章于 2025-06-28 09:57:16 发布
阅读量1.4k 收藏 12
点赞数 12
CC 4.0 BY-SA版权
文章标签： linux debian ubuntu RKNN
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/gaoxukkk888/article/details/139518115

RKNN-Toolkit2不同版本导出的RKNN模型输出不同

在我的虚拟机中安装过RKNN-Toolkit2的1.4.0版本
在这里插入图片描述

在转换RockChip官方提供的的RetinaFace_mobile320.onnx模型(下载链接为：https://ftrg.zbox.filez.com/v2/delivery/data/95f00b0fc900458ba134f8b180b3f7a1/examples/RetinaFace/RetinaFace_mobile320.onnx)为RKNN后，查看网络输出如下：

在这里插入图片描述

可以看到RKNN-Toolkit2的1.4.0转换的RetinaFace_mobile320.onnx的三个输出都是四维的，如果使用这个RKNN模型和RockChip官方提供的后处理代码去RK3588上进行推理会报错，显示维度问题。

分析官方的后处理代码和原模型的推理代码发现三个输出应该是三维的，接着查找相关资料发现别人的RetinaFace推理没遇到问题，最后怀疑是RKNN-Toolkit2的版本转换出来的RKNN模型推理的结果不一样。

卸载RKNN-Toolkit2的1.4.0，安装最新的RKNN-Toolkit2版本时出现过下面这种情况（python=3.8）

在这里插入图片描述

开始我以为时是现在

最低0.47元/天 解锁文章

200万优质内容无限畅学
【RKNN】RKNN-Toolkit2 Python 模型加载API函数详解
浩瀚之水的专栏
05-02 125
RKNN-Toolkit2 提供多种函数加载不同框架的预训练模型，将其转换为 RKNN 格式。
【RKNN】RKNN-Toolkit2 Python API函数详解
浩瀚之水的专栏
04-29 221
RKNN-Toolkit2 的 API 设计以开发者效率为核心，覆盖模型转换、量化、部署全流程。通过灵活的参数配置（如多核调度、混合量化）和调试工具（性能分析、精度对比），可快速适配复杂边缘场景。
瑞芯微转化人脸检测retinaface模型
ZuoSeDiao的博客
04-18 2209
瑞芯微转化人脸检测retinaface模型一、运行docker二、转换步骤1.使用https://netron.app/查看模型的输入及输出2.设置转换模型参数3.运行文件生成rknn模型文件3.调用模型测试总结 一、运行docker docker run -t -i --privileged -v /dev/bus/usb:/dev/bus/usb rknn-toolkit:1.7.1 /bin/bash 二、转换步骤 1.使用https://netron.app/查看模型的输入及输出 2.设置转换模
RKNN Toolkit2 工具的使用
qq_41808475的博客
04-28 2407
RKNN Toolkit2 是由瑞芯微电子 (Rockchip) 开发的一套用于深度学习模型优化和推理的工具。它主要面向在瑞芯微SoC上进行AI应用开发，但也可以用于PC平台进行模型的转换、量化、推理等操作。它支持将多种深度学习框架的模型（如Caffe, TensorFlow, PyTorch等）转换为RKNN格式。
【RKNN】RKNN-Toolkit2 Python API之rknn_outputs_get 函数详解
最新发布
浩瀚之水的专栏
06-28 997
rknn_outputs_get 是 RKNN-Toolkit2 中用于 ​获取模型推理结果​ 的核心接口，支持多输出模型的解析与后处理。
RKNN 和rknn-toolkit2
xie__jin__cheng的博客
09-13 1684
RKNN 和rknn-toolkit2
模型部署——rknn-toolkit-lite2部署RKNN模型到开发板上（python版）
qq_40280673的博客
03-07 1万+
rknn-toolkit-lite2部署RKNN模型到开发板上
【rknn】onnx转rknn脚本解读以及函数解读(版本V1.7.3)
sazass的博客
02-22 7000
创建rknn对象。rknn的参数配置，这个函数需要设置的参数细节很多。加载onnx模型构建rknn模型模型转换并输出rknn模型接下来，依次具体介绍上述这些函数，以及包含的所有参数和默认值。建议直接在官网github上找到最新的整体流程图：当前介绍的rknn-api版本是V1.7.3。
rknn-toolkit-x86-cp35.whl
11-02
因为官方的下载已经不支持python3.5的版本，所以上传rknn-toolkit-cp35版本，以便于像我一样的使用python3.5的用户在PC安装虚拟环境
【环境搭建paddle转onnx工具包】1.6.0版本rknn-toolkit2.zip
02-26
2. **模型转换**：使用`RKNN-Toolkit2`将ONNX模型转换为RKNN模型。这个过程通常涉及模型的优化和量化，以适应低功耗设备的计算需求。命令行工具如`rknn_convert`可以帮助完成此步骤。 3. **模型验证**：转换后，可以...
【RKNN】RKNN-Toolkit2 Python API之export_rknn函数详解
浩瀚之水的专栏
05-03 62
export_rknn() 是 RKNN-Toolkit2 中用于 ​保存转换后的 RKNN 模型​ 的关键方法，支持导出标准模型或预编译模型。
基于RKNN的深度神经网络模型设计建议1
08-03
1. 卷积核设置 2. 融合结构设计 3. 关于 2D 卷积和 Depthwise 卷积的使用 4. 网络稀疏化
pytorch模型转ONNX模型(支持批处理)
热门推荐
zxgmlcj的博客
11-27 2万+
本篇幅介绍pytorch模型转ONNX模型 一、pytorch模型保存/加载 有两种方式可用于保存/加载pytorch模型 1）文件中保存模型结构和权重参数 2）文件只保留模型权重. 1、文件中保存模型结构和权重参数 1）pytorch模型保存 import torch torch.save(selfmodel,"save.pt") 2）pytorch模型加载 import torch torc...
14. rk3588自带的RKNNLite检测yolo模型（python)
weixin_71719718的博客
02-20 2381
大家对以上代码是不是很熟悉，其实这部分代码的很多函数都是将onnx文件转换为rknn文件的代码，只是把它的from rknn.api import RKNN修改为from rknnlite.api import RKNNLite，并且将载入rknn模型的内容修改即可，数据的处理完全没变。用这个代码的好处是，我们以后对相机进行检测目标时，只需要下载一次rknn模型，后续只需要处理图片即可。这里没有出车，是我训练的模型问题。
模型部署——RKNN模型的评估和推理测试（附代码）
qq_40280673的博客
02-25 1万+
RKNN模型的评估和推理测试
【Ubantu版本：18.04 RKNN版本：1.14.0 cuda版本：9.0】
xiaommianyang的博客
04-11 561
【Ubantu版本：18.04 RKNN版本：1.14.0 cuda版本：9.0】安装以及转换过程记录。 环境的安装：之前安装的版本是根据给出的安装说明书里的步骤安装的rknn1.3.0，但是会报错如下：ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory 这就显示该环境下你的CUDA的版本和你的TF版本是不匹配的，需要该TF版本安装CUDA为9.0的环境。ubantu可以自己创建软
NPU 驱动版本、rknn_server 版本、librknn_runtime 版本以及 RKNN Toolkit 版本的对应关系
m0_63474848的博客
04-11 1054
NPU的驱动在下载地址中的/drivers/目录下。npu_ko：NPU内核驱动KO。
【RK3588 第四篇】YOLO V5中pytorch2onnx，pytorch和onnx模型输出不一致，精度降低
钱多多先森
10-13 5326
pytorch转onnx模型发现的问题，以及转rknn出现的精度下降，同时验证了新的pytorch转onnx模型出现的麻烦事
rknn如何查看runtime和驱动的版本
重剑无锋博客
08-24 3000
【代码】rknn如何查看runtime和驱动的版本。
rknn-toolkit2和
04-05
RKNN-Toolkit2 主要功能包括模型加载、量化处理以及导出为 RKNN 格式的文件。下面是一个简单的例子展示如何将 TensorFlow 模型转换成 RKNN 格式： ```python from rknn.api import RKNN # 创建 RKNN 对象实例 rknn...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值