Toybrick

pytorch 转 onnx 再转 rknn 报错

kkkaaa

中级会员

积分
203
楼主
发表于 2020-4-21 16:49:30    查看: 5519|回复: 2 | [复制链接]    打印 | 只看该作者
从 pytorch 转出来的 onnx 模型通过 onnx.checker.check_model(onnx_model) 没有报错
但是把 onnx 模型转 rknn 模型时报错:
E Catch exception when loading onnx model: /data01/wens/workspace/project/Yet-Another-EfficientDet-Pytorch/weights/efficientdet_d0.onnx!
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 510, in rknn.api.rknn_base.RKNNBase.load_onnx
E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 494, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.__init__
E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 504, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.__init__
E   File "rknn/base/RKNNlib/converter/onnx_util.py", line 158, in rknn.base.RKNNlib.converter.onnx_util.ONNXProto_Util.polishing_model
E   File "/data01/wens/venv/rknn/lib/python3.6/site-packages/onnx/utils.py", line 18, in polish_model
E     onnx.checker.check_model(model)
E   File "/data01/wens/venv/rknn/lib/python3.6/site-packages/onnx/checker.py", line 86, in check_model
E     C.check_model(model.SerializeToString())
E onnx.onnx_cpp2py_export.checker.ValidationError: Field 'shape' of type is required but missing.

请问这是不是因为转的 onnx 模型的问题?


回复

使用道具 举报

jefferyzhang

版主

积分
12956
沙发
发表于 2020-4-21 16:56:35 | 只看该作者
onnx模型先用onnx加载下看是否能正常推理。。
回复

使用道具 举报

kkkaaa

中级会员

积分
203
板凳
 楼主| 发表于 2020-4-21 17:10:03 | 只看该作者
onnxruntime.InferenceSession 的确报错了
回复

使用道具 举报

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

本版积分规则

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


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