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

固件类型:官方提供的固件
固件文件名称: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也能检测出目标,所以不存在该方面问题。
tchip_askquestions




none.txt (3 Bytes)

你好,可以尝试更新1.4版本的工具试一下。https://github.com/rockchip-linux/rknn-toolkit2/blob/master/doc/changelog-1.4.0.txt

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

可以进Q群1025468710交流一下