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

[复制链接] · 发表于 2022-3-29 14:54:09

尝试用自己训练的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.

只看该作者 · 发表于 2024-3-7 15:10:59

你好，请问你这个问题解决了吗？w我也遇到一样的问题

只看该作者 · 发表于 2024-3-7 15:47:30

不要把nms 这些后处理操作转换在onnx中（也就是官方yolov5的自动转换），应该是3个输出头，要做修改的。

只看该作者 · 发表于 2024-6-24 15:22:04

aurora 发表于 2024-3-7 15:47
不要把nms 这些后处理操作转换在onnx中（也就是官方yolov5的自动转换），应该是3个输出头，要做修改的。
...

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

只看该作者 · 发表于 2024-6-24 15:45:02

1569063835 发表于 2024-6-24 15:22
你好，请问这个应该做怎样修改呢，我还是没有解决

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

只看该作者 · 发表于 2024-6-24 15:59:51

hushp3 发表于 2024-3-7 15:10
你好，请问你这个问题解决了吗？w我也遇到一样的问题

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

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

本帖子中包含更多资源