New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

#

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Jump to bottom

Support static model export with End2End (EfficientNMS) #479

Open

laugh12321 wants to merge 1 commit into WongKinYiu:main from laugh12321:main

laugh12321 commented Jun 9, 2024

The previous onnx_end2end option enforced dynamic model export, preventing static model export.

This pull request introduces the ability to export static models.

Example:

# dynamic export
python3 export.py --weights ./yolov9-c.pt --imgsz 640 --simplify --include onnx_end2end --dynamic

# static export with batch size 2
python3 export.py --weights ./yolov9-c.pt --imgsz 640 --simplify --include onnx_end2end --batch-size 2


          Support static model export with End2End (EfficientNMS)

43b624b

# for free to join this conversation on GitHub. Already have an account? # to comment

Labels

None yet