Firefly开源社区

打印 上一主题 下一主题

[技术讨论] ONNX转RKNN时报错,是算子不支持吗

44

积分

0

威望

0

贡献

技术小白

积分
44

ONNX转RKNN时报错,是算子不支持吗

发表于 2022-9-21 16:09:47      浏览:5427 | 回复:3        打印      只看该作者   [复制链接] 楼主
  • 固件类型: 官方提供的固件
  • 固件文件名称: ITX-3588J_Ubuntu20.04-Gnome-r21199_v1.0.1b_220812.
  • 固件下载地址: 官方下载地址
  • Log日志: none.txt
问题描述及复现步骤:
onnx转rknn时出现转换失败的问题,附图是自己改写的onnx2rknn源码、ONNX模型的输入输出信息、模型转换报错信息。不知道是不是官方算子不支持还是其它什么原因,有没有大佬指导一下;

PS:因为是在PC端进行的onnx转rknn,并无日志信息,忽略none.txt吧。ONNX模型来自于自己训练的一个图像拼接算法,并非是官方toolkit onnx demo里的yolov5或resnet,附上github源码地址:[url=]https://github.com/Kolkir/Coarse_LoFTR_TRT/tree/main/weights[/url]

背景:ONNX模型是pytorch模型转换来的,直接使用onnxruntime是可以跑通输出的,确保ONNX本身应该没有问题,也尝试过了onnx_simplify简化模型,转rknn时仍旧报一样的错误。rknn_toolkit2本身也没有任何安装或者环境问题,因为自己在做yolov5转rknn时非常顺利,得到的yolov5.rknn也能检测出目标,所以不存在该方面问题。

code.png (60.28 KB, 下载次数: 609)

code.png

onnx_describe.png (19.45 KB, 下载次数: 581)

onnx_describe.png

wrong_info.png (20.71 KB, 下载次数: 572)

wrong_info.png

none.txt

3 Bytes, 下载次数: 0, 下载积分: 灯泡 -1 , 经验 -1

回复

使用道具 举报

1万

积分

18

威望

20

贡献

神级创客

Rank: 8Rank: 8

积分
10851

突出贡献

发表于 2022-9-22 09:05:38        只看该作者  沙发
你好,可以尝试更新1.4版本的工具试一下。https://github.com/rockchip-linu ... changelog-1.4.0.txt
回复

使用道具 举报

44

积分

0

威望

0

贡献

技术小白

积分
44
发表于 2022-9-22 19:05:37        只看该作者  板凳
zyk 发表于 2022-9-22 09:05
你好,可以尝试更新1.4版本的工具试一下。https://github.com/rockchip-linux/rknn-toolkit2/blob/master/d ...

安装了1.4版本的工具后,并未解决该问题,还是一样的报错Meet unsupported dims in reducesum, dims: 5
回复

使用道具 举报

1万

积分

18

威望

20

贡献

神级创客

Rank: 8Rank: 8

积分
10851

突出贡献

发表于 2022-9-23 18:15:33        只看该作者  地板
可以进Q群1025468710交流一下
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

友情链接 : 爱板网 电子发烧友论坛 云汉电子社区 粤ICP备14022046号-2
快速回复 返回顶部 返回列表