Toybrick

标题: onnx转rknn遇到以下问题 [打印本页]

作者: 自由的学习    时间: 2023-8-17 10:25
标题: onnx转rknn遇到以下问题

D Calc tensor Initializer_704 (64,)
D Calc tensor Initializer_703 (64, 3, 7, 7)
E Calc node Conv : Conv_0 output shape fail
E Catch exception when loading onnx model: pose_best_230811.pth.onnx!
E Traceback (most recent call last):
E   File "rknn/base/RKNNlib/onnx_ir/onnx_numpy_backend/shape_inference.py", line 65, in rknn.base.RKNNlib.onnx_ir.onnx_numpy_backend.shape_inference.infer_shape
E   File "rknn/base/RKNNlib/onnx_ir/onnx_numpy_backend/ops/conv.py", line 40, in rknn.base.RKNNlib.onnx_ir.onnx_numpy_backend.ops.conv.Conv
E   File "rknn/base/RKNNlib/onnx_ir/onnx_numpy_backend/ops/max_pool.py", line 50, in rknn.base.RKNNlib.onnx_ir.onnx_numpy_backend.ops.max_pool.get_output_shape
E IndexError: list index out of range
E During handling of the above exception, another exception occurred:
E Traceback (most recent call last):
E   File "rknn/base/RKNNlib/app/importer/import_onnx.py", line 120, in rknn.base.RKNNlib.app.importer.import_onnx.Importonnx.run
E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 123, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.__init__
E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 1187, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx._shape_inference
E   File "rknn/base/RKNNlib/onnx_ir/onnx_numpy_backend/shape_inference.py", line 70, in rknn.base.RKNNlib.onnx_ir.onnx_numpy_backend.shape_inference.infer_shape
E   File "rknn/api/rknn_log.py", line 323, in rknn.api.rknn_log.RKNNLog.e
E ValueError: Calc node Conv : Conv_0 output shape fail
E Please feedback the detailed log file <log_feedback_to_the_rknn_toolkit_dev_team.log> to the RKNN Toolkit development team.


上面是log里面的记录,转换过程加载完onnx,在build rknn过程中就突然中断了,下面是运行截图,请问是什么原因导致的呢,刚接触这个也不是很懂,之前yolov5转是成功的,这个是自己的一个模型


作者: jefferyzhang    时间: 2023-8-17 17:30
rknn-toolkit版本号、onnx opset版本号 分别是多少
作者: 自由的学习    时间: 2023-8-17 19:09
jefferyzhang 发表于 2023-8-17 17:30
rknn-toolkit版本号、onnx opset版本号 分别是多少

rknn-toolkit 是1.7.5  onnx是1.10.0  onnxoptimizer是0.2.7
作者: jefferyzhang    时间: 2023-8-18 09:31
opset,不是问你onnx版本号




欢迎光临 Toybrick (https://t.rock-chips.com/) Powered by Discuz! X3.3