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

Meeeteor · 发表于 2022-9-21 16:09:47

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也能检测出目标，所以不存在该方面问题。

zyk · 发表于 2022-9-22 09:05:38

你好，可以尝试更新1.4版本的工具试一下。https://github.com/rockchip-linu ... changelog-1.4.0.txt

Meeeteor · 发表于 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

zyk · 发表于 2022-9-23 18:15:33

可以进Q群1025468710交流一下

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

【技术讨论】 ONNX转RKNN时报错，是算子不支持吗

突出贡献