ultralytics / ultralytics

NEW - YOLOv8 πŸš€ in PyTorch > ONNX > OpenVINO > CoreML > TFLite
https://docs.ultralytics.com
GNU Affero General Public License v3.0
23.76k stars 4.74k forks source link

New ONNX, TorchScript, CoreML export test matrices #11652

Closed glenn-jocher closed 1 week ago

glenn-jocher commented 1 week ago

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhancements in YOLO model export functionality across formats including TorchScript, ONNX, OpenVINO, CoreML, TFLite, *.pb, Paddle, and NCNN.

πŸ“Š Key Changes

🎯 Purpose & Impact

codecov[bot] commented 1 week ago

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Project coverage is 70.69%. Comparing base (2583f84) to head (9ba2ab4).

:exclamation: Current head 9ba2ab4 differs from pull request most recent head 57a01df. Consider uploading reports for the commit 57a01df to get more accurate results

Additional details and impacted files ```diff @@ Coverage Diff @@ ## main #11652 +/- ## ========================================== + Coverage 70.60% 70.69% +0.08% ========================================== Files 122 122 Lines 15583 15584 +1 ========================================== + Hits 11003 11017 +14 + Misses 4580 4567 -13 ``` | [Flag](https://app.codecov.io/gh/ultralytics/ultralytics/pull/11652/flags?src=pr&el=flags&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=ultralytics) | Coverage Ξ” | | |---|---|---| | [Benchmarks](https://app.codecov.io/gh/ultralytics/ultralytics/pull/11652/flags?src=pr&el=flag&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=ultralytics) | `35.67% <100.00%> (+0.06%)` | :arrow_up: | | [GPU](https://app.codecov.io/gh/ultralytics/ultralytics/pull/11652/flags?src=pr&el=flag&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=ultralytics) | `37.37% <0.00%> (-0.01%)` | :arrow_down: | | [Tests](https://app.codecov.io/gh/ultralytics/ultralytics/pull/11652/flags?src=pr&el=flag&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=ultralytics) | `66.78% <100.00%> (+0.06%)` | :arrow_up: | Flags with carried forward coverage won't be shown. [Click here](https://docs.codecov.io/docs/carryforward-flags?utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=ultralytics#carryforward-flags-in-the-pull-request-comment) to find out more.

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.

glenn-jocher commented 1 week ago

@Burhan-Q trying this out now that it's working well for OpenVINO. ONNX is trickier though, FP16 export is only supported on CUDA as far as I know, and we lack INT8 support, so the matrix is a little less fun.

Torchscript even more so.