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, 下载次数: 625)
-
onnx_describe.png (19.45 KB, 下载次数: 591)
-
wrong_info.png (20.71 KB, 下载次数: 584)
-
3 Bytes, 下载次数: 0, 下载积分: 灯泡 -1 , 经验 -1