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个输出头，要做修改的。

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