khadas / vim4_npu_applications

0 stars 2 forks source link

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

Open AatishLanghee opened 1 year ago

AatishLanghee commented 1 year ago

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]