Toybrick

标题: onnx转rknn时：Field 'shape' of type is required but missing. [打印本页]

作者: Robin 时间: 2022-3-29 14:54
标题: onnx转rknn时：Field 'shape' of type is required but missing.
尝试用自己训练的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.

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

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

作者: 1569063835 时间: 2024-6-24 15:22

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

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

作者: jefferyzhang 时间: 2024-6-24 15:45

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

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

作者: 1569063835 时间: 2024-6-24 15:59

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

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

欢迎光临 Toybrick (https://t.rock-chips.com/)