Skip to content
New issue

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

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

Already on GitHub? Sign in to your account

Add --int8 argument #4799

Merged
merged 3 commits into from
Sep 15, 2021
Merged

Add --int8 argument #4799

merged 3 commits into from
Sep 15, 2021

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Sep 15, 2021

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhancements to model export functionality with INT8 quantization support for TensorFlow Lite. πŸš€

πŸ“Š Key Changes

  • Avoids adding ROOT to sys.path if it's already present.
  • Renames the tfl_int8 parameter to int8 for clarity.
  • Adds INT8 quantization support for CoreML and TensorFlow Lite model exports.

🎯 Purpose & Impact

  • Prevents unnecessary duplication in sys.path, leading to cleaner code execution.
  • The clearer parameter naming (int8) improves code readability and understanding.
  • Enables users to export TensorFlow Lite models with INT8 quantization, potentially reducing model size and improving performance on INT8-capable hardware. This change expands the utility of YOLOv5 for mobile and edge devices with optimized inference capabilities. πŸ‘πŸ’¨

@glenn-jocher glenn-jocher merged commit b74dd4b into master Sep 15, 2021
@glenn-jocher glenn-jocher deleted the update/export branch September 15, 2021 09:33
CesarBazanAV pushed a commit to CesarBazanAV/yolov5 that referenced this pull request Sep 29, 2021
* Add `--int8` argument

* parents[0] bug fix

* Fix order
BjarneKuehl pushed a commit to fhkiel-mlaip/yolov5 that referenced this pull request Aug 26, 2022
* Add `--int8` argument

* parents[0] bug fix

* Fix order
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant