How to export to ONNX? - Githubissues

huawei-noah / Efficient-Computing

Efficient computing methods developed by Huawei Noah's Ark Lab

1.21k stars 211 forks source link

How to export to ONNX? #78

Closed Egorundel closed 10 months ago

Egorundel commented 1 year ago

Hello!

How to export the received weights to ONNX model with dynamic batch size and with NMS for further conversion of ONNX to TensorRT for inference?

lose4578 commented 1 year ago

Hi, we didn't test the dynamic batch for ONNX in our code. Our code follow in YOLOv6, and maybe you can find a solution in https://github.com/meituan/YOLOv6.