Firefly开源社区

标题: ONNX转RKNN时报错，是算子不支持吗 [打印本页]

作者: Meeeteor 时间: 2022-9-21 16:09
标题: ONNX转RKNN时报错，是算子不支持吗
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也能检测出目标，所以不存在该方面问题。
tchip_askquestions

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

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

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

none.txt

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

作者: zyk 时间: 2022-9-22 09:05
你好，可以尝试更新1.4版本的工具试一下。https://github.com/rockchip-linu ... changelog-1.4.0.txt

作者: Meeeteor 时间: 2022-9-22 19:05

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

作者: zyk 时间: 2022-9-23 18:15
可以进Q群1025468710交流一下

欢迎光临 Firefly开源社区 (https://dev.t-firefly.com/)