Toybrick

onnx转rknn时:Field 'shape' of type is required but missing.

Robin

新手上路

积分
5
楼主
发表于 2022-3-29 14:54:09    查看: 3275|回复: 2 | [复制链接]    打印 | 只看该作者
尝试用自己训练的YOLO模型跑在RK3568上,跑通了DEMO的yolov5s.onnx转rknn,并在板子上实现推理。但是换自己的onnx模型都不行,怀疑自己的pytorch转onnx不对,我又用了官方yolov5的自带转换,转换了现成的yolov5s.pt,还是报错一样的错误。求大佬指点。
E load_onnx:     C.check_model(model.SerializeToString())

E load_onnx: onnx.onnx_cpp2py_export.checker.ValidationError: Field 'shape' of type is required but missing.


本帖子中包含更多资源

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

x
回复

使用道具 举报

hushp3

新手上路

积分
6
沙发
发表于 2024-3-7 15:10:59 | 只看该作者
你好,请问你这个问题解决了吗?w我也遇到一样的问题
回复

使用道具 举报

aurora

中级会员

积分
461
板凳
发表于 2024-3-7 15:47:30 | 只看该作者
不要把nms 这些后处理操作转换在onnx中(也就是官方yolov5的自动转换),应该是3个输出头,要做修改的。
回复

使用道具 举报

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

本版积分规则

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


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