how to export the onnx model ?

[xinsuinizhuan]

❔Question

I export the model with: python models/export.py --weights Yolov5s6_pose_960.pt --img 960 --batch 1 --simplify --export-nms

the model with output is:

and export with no nms is: so many output layers

how to export with the output:

Additional context

[xinsuinizhuan]

when i export with: python models/export.py --weights Yolov5s6_pose_960.pt --img 960 --batch 1 --simplify the result is:

[PredyDaddy]

Hey, I am looking at this repo. Could you add my wechat to talk together about how to deploy this to TensorRT and Deepstream. My wechat account is 1642942273