Toybrick

模型转换load onnx报错

jasonlyo

新手上路

积分
42
楼主
发表于 6 天前    查看: 138|回复: 1 | [复制链接]    打印 | 只看该作者
onnx模型转rknn,报错如下,请问是模型有啥问题吗?RKNN-toolkit版本是1.7.1。详细的日志,可以看附件。模型由于大小限制暂时传不上来。

--> Loading model
W Call onnx.optimizer.optimize fail, skip optimize
/mnt/python36new/lib/python3.6/site-packages/rknn/api/rknn.py:150: RuntimeWarning: overflow encountered in exp
  ret = self.rknn_base.load_onnx(model, inputs, input_size_list, outputs)
E Calc node Slice : /model.22/Slice output shape fail
E Catch exception when loading onnx model: yolov8n_opset11_imgsz2560_fp32_simplifyFalse_dynamicFalse_optimizeFalse.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/slice.py", line 41, in rknn.base.RKNNlib.onnx_ir.onnx_numpy_backend.ops.slice.Slice
E   File "<string>", line 1, in <module>
E TypeError: slice indices must be integers or None or have an __index__ method
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 118, in rknn.base.RKNNlib.app.importer.import_onnx.Importonnx.run
E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 112, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.__init__
E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 1062, 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 Slice : /model.22/Slice 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.
E You can also check github issues: https://github.com/rockchip-linux/rknn-toolkit/issues



本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

andornot

注册会员

积分
182
沙发
发表于 5 天前 | 只看该作者
yolov8官方模型转rknn时,output部分有算子不支持,
你看下rknn_model_zoo中yolov8的example,有对output作修改优化,
文档地址: https://github.com/airockchip/rk ... ain/examples/yolov8
回复

使用道具 举报

andornot

注册会员

积分
182
板凳
发表于 5 天前 | 只看该作者
yolov8官方模型,output部分有算子不支持,
你看下rknn_model_zoo中有 yolov8的example,有对output做了修改优化,
文档地址: https://github.com/airockchip/rk ... ain/examples/yolov8
回复

使用道具 举报

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

本版积分规则

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


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