NVIDIA-AI-IOT / yolo_deepstream

yolo model qat and deploy with deepstream&tensorrt
Apache License 2.0
550 stars 139 forks source link

Efficient NMS option exporting to ONNX #33

Closed labar90 closed 1 year ago

labar90 commented 1 year ago

Is it possible to export to onnx with the option of efficient NMS in the quantization code?

wanghr323 commented 1 year ago

Is it possible to export to onnx with the option of efficient NMS in the quantization code?

Yes, It is possible, But we did not do that, the origin yolov7 repo did that, you can control with the export flag. ref: https://github.com/WongKinYiu/yolov7#export

I will close this ISSUE, If you have any further questions, pls reopen the ISSUE

Thanks Haoran Wang