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

feature/add_tf_and_tflite_export_2 #4449

Closed
wants to merge 3 commits into from

Conversation

SkalskiP
Copy link
Contributor

@SkalskiP SkalskiP commented Aug 17, 2021

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Adding explicit TFLite model support and TensorFlow export functionality to Ultralytics' YOLOv5.

πŸ“Š Key Changes

  • Added TFLite and TensorFlow *.pb model extension support in .gitignore.
  • Provided an option to disable layer fusion during model loading in attempt_load function for better compatibility with TensorFlow's layer behavior.
  • Implemented an extensive new tf.py script to facilitate TensorFlow and TFLite model conversion, support, and inference operations for YOLOv5 models.
  • Augmented the requirements.txt with a specific TensorFlow version required for TFLite export (tensorflow==2.4.1).

🎯 Purpose & Impact

  • πŸš€ Enabling broader adoption: With the ability to convert YOLOv5 models to TensorFlow and TFLite formats, developers can now deploy YOLOv5 on a broader range of platforms and devices, including those supporting TensorFlow Lite.
  • πŸ”Œ Greater flexibility: Users can now choose to load models with or without layer fusion, providing more control based on different deployment needs.
  • 🧠 Facilitating quantization: TFLite export supports FP16 and Int8 quantization, thereby enabling efficient deployment on edge devices with limited computational resources.
  • πŸ€– Broader inference possibilities: Detecting objects can now be performed using TensorFlow models, leveraging its vast ecosystem.
  • 🏹 Streamlined workflow: By integrating conversion scripts and compatibility changes directly into YOLOv5's repository, the process of taking a model from training to deployment on TensorFlow-compatible environments is greatly simplified.

@glenn-jocher
Copy link
Member

@SkalskiP let's close this PR and delete this branch as #4479 has taken over this task now. Thanks!

@glenn-jocher glenn-jocher deleted the feature/add_tf_and_tflite_export_2 branch September 18, 2021 17:39
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

2 participants