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

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)

在这里插入图片描述

开始我以为时是现在库都更新了,各种库的版本和python版本不兼容了,引发的冲突

直到我看到小诺最帅气分享的https://blog.csdn.net/weixin_51447731/article/details/137088316?spm=1001.2014.3001.5506文章,发现原来是清华源暂时没有tf-estimator-nightly==2.8.0.dev2021122109这个安装包

故我执行下面语句单独安装这个包

pip install -i https://pypi.doubanio.com/simple/ tf-estimator-nightly==2.8.0.dev2021122109

然后安装requirements中的库和新版的RKNN-Toolkit2,成功安装

在这里插入图片描述

使用新版本的RKNN-Toolkit2转换RetinaFace_mobile320.onnx模型,观察模型

在这里插入图片描述

此时的RKNN模型的三个输出为三维了,RKNN-Toolkit2的不同版本转换出来的RKNN模型推理的结果是不一样的。

相关推荐

  1. 使用rknn-toolkit2将paddleseg模型导出rknn模型

    2024-06-07 12:02:01       65 阅读
  2. [python][whl]rknn_toolkit_lite2whl版本下载地址汇总

    2024-06-07 12:02:01       33 阅读
  3. RK3588 rknpu2rknn-toolkit2使用说明

    2024-06-07 12:02:01       97 阅读
  4. orangepi-5b 使用 rknn-toolkit2 实测

    2024-06-07 12:02:01       33 阅读
  5. [笔记] rknn Toolkit1.6=>2.0 API变动注释

    2024-06-07 12:02:01       22 阅读
  6. RKNN Toolkit Lite2 一键安装和测试,sh脚本

    2024-06-07 12:02:01       65 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-06-07 12:02:01       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-06-07 12:02:01       101 阅读
  3. 在Django里面运行非项目文件

    2024-06-07 12:02:01       82 阅读
  4. Python语言-面向对象

    2024-06-07 12:02:01       91 阅读

热门阅读

  1. Web前端ES6-ES13笔记合集

    2024-06-07 12:02:01       23 阅读
  2. mongodb集群

    2024-06-07 12:02:01       27 阅读
  3. MongoDB $rename 给字段一次重新命名的机会

    2024-06-07 12:02:01       31 阅读
  4. docker-compose部署Redis6.0

    2024-06-07 12:02:01       22 阅读
  5. Spark read load Parquet Files

    2024-06-07 12:02:01       28 阅读
  6. 悉数六大设计原则

    2024-06-07 12:02:01       19 阅读
  7. 超长正整数的加法

    2024-06-07 12:02:01       30 阅读