How do we export Yolo v8n ONNX model with required output shape.

[AatishLanghee]

I want to know how can I export the yolo v8n onnx model into the below format :

images
name: images
type: float32[1,3,640,640]

output0
name: output0
type: float32[1,80,80,144]
342
name: 342
type: float32[1,40,40,144]
358
name: 358
type: float32[1,20,20,144]

Currently the output shape is coming like this:

images
name: images
type: float32[1,3,640,640]

output0
name: output0
type: float32[1,84,8400]