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

TFLite test matrix #13083

Closed
wants to merge 15 commits into from
Closed

TFLite test matrix #13083

wants to merge 15 commits into from

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented May 23, 2024

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhanced TensorFlow TFLite export capabilities and resolved TensorFlow installation conflicts.

πŸ“Š Key Changes

  • Implemented tests for exporting YOLO models to TFLite format, ensuring compatibility and addressing performance.
  • Updated TensorFlow dependency requirements, removing specific version constraints, and added Keras as a separate requirement.
  • Included conditional requirements to cater for different platforms and hardware configurations, improving installation success rates.
  • Excluded test cases with incompatible settings (i.e., both int8 and half precision enabled) for better clarity and reliability.

🎯 Purpose & Impact

  • Clarifies TensorFlow Installations: By specifying TensorFlow and Keras as separate requirements, users should experience fewer conflicts during installation, especially on Windows and macOS πŸ–₯️🍏.
  • Enhances Model Exporting: The new tests and refinements aim to make exporting models to TFLite format smoother, potentially leading to easier deployment on mobile and edge devices πŸ“±πŸš€.
  • Increases Compatibility: Adjusting dependencies based on platform and hardware ensures more users can successfully set up their environment without diving into dependency resolution issues 🧩.
  • Improves Test Coverage and Reliability: By focusing on realistic test scenarios and preventing incompatible configurations, the quality and predictability of exports are enhanced βœ….

Overall, these changes aim to make the process of using TensorFlow with Ultralytics projects more intuitive and trouble-free for developers and enthusiasts alike, promoting broader use and application of machine learning models 🌐.

Copy link

codecov bot commented May 23, 2024

Codecov Report

Attention: Patch coverage is 0% with 1 lines in your changes are missing coverage. Please review.

Project coverage is 69.43%. Comparing base (8caa1c9) to head (f8d4f8b).

Files Patch % Lines
ultralytics/engine/exporter.py 0.00% 1 Missing ⚠️
Additional details and impacted files
@@            Coverage Diff             @@
##             main   #13083      +/-   ##
==========================================
- Coverage   69.67%   69.43%   -0.24%     
==========================================
  Files         124      124              
  Lines       15655    15655              
==========================================
- Hits        10907    10870      -37     
- Misses       4748     4785      +37     
Flag Coverage Ξ”
Benchmarks 35.12% <0.00%> (-0.02%) ⬇️
GPU 37.27% <0.00%> (ΓΈ)
Tests 64.89% <0.00%> (-1.03%) ⬇️

Flags with carried forward coverage won't be shown. Click here to find out more.

β˜” View full report in Codecov by Sentry.
πŸ“’ Have feedback on the report? Share it here.

@glenn-jocher glenn-jocher added the TODO Items that needs completing label May 26, 2024
@glenn-jocher glenn-jocher deleted the test-tflite branch May 29, 2024 08:38
@glenn-jocher
Copy link
Member Author

Closed in favor of #13176

@glenn-jocher glenn-jocher removed the TODO Items that needs completing label May 29, 2024
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.

None yet

1 participant