Toybrick

onnx转rknn遇到以下问题

自由的学习

注册会员

积分
172
发表于 2023-8-17 10:25:54    查看: 2412|回复: 3 | [复制链接]    打印 | 显示全部楼层

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

版主

积分
13405
发表于 2023-8-17 17:30:24 | 显示全部楼层
rknn-toolkit版本号、onnx opset版本号 分别是多少
回复

使用道具 举报

自由的学习

注册会员

积分
172
 楼主| 发表于 2023-8-17 19:09:08 | 显示全部楼层
jefferyzhang 发表于 2023-8-17 17:30
rknn-toolkit版本号、onnx opset版本号 分别是多少

rknn-toolkit 是1.7.5  onnx是1.10.0  onnxoptimizer是0.2.7
回复

使用道具 举报

jefferyzhang

版主

积分
13405
发表于 2023-8-18 09:31:26 | 显示全部楼层
opset,不是问你onnx版本号
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表