Toybrick

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

Robin

新手上路

积分
5
发表于 2022-3-29 14:54:09    查看: 6097|回复: 5 | [复制链接]    打印 | 显示全部楼层
尝试用自己训练的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

高级会员

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

使用道具 举报

1569063835

新手上路

积分
30
发表于 2024-6-24 15:22:04 | 显示全部楼层
aurora 发表于 2024-3-7 15:47
不要把nms 这些后处理操作转换在onnx中(也就是官方yolov5的自动转换),应该是3个输出头,要做修改的。
...

你好,请问这个应该做怎样修改呢,我还是没有解决
回复

使用道具 举报

jefferyzhang

版主

积分
13608
发表于 2024-6-24 15:45:02 | 显示全部楼层
1569063835 发表于 2024-6-24 15:22
你好,请问这个应该做怎样修改呢,我还是没有解决

官方model zone有示例,可以自行研究下.
https://github.com/airockchip
回复

使用道具 举报

1569063835

新手上路

积分
30
发表于 2024-6-24 15:59:51 | 显示全部楼层
hushp3 发表于 2024-3-7 15:10
你好,请问你这个问题解决了吗?w我也遇到一样的问题

你好,请问你的问题解决了吗,我也遇到了这个问题
回复

使用道具 举报

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

本版积分规则

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


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