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 --keras argument for TF exports #7921

Merged
merged 1 commit into from
May 22, 2022
Merged

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented May 22, 2022

Resolves #7911 (comment)

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhanced export capabilities with Keras model support in YOLOv5.

πŸ“Š Key Changes

  • Added a keras flag to enable Keras model export functionality.
  • Incorporated Keras export logic into the model export process.
  • Updated command-line argument parser to accept the new --keras parameter for triggering Keras exports.

🎯 Purpose & Impact

  • Purpose: This change allows users to export YOLOv5 models as Keras models, providing additional flexibility.
  • Impact: Users can now integrate YOLOv5 with applications and systems that operate better with Keras models, potentially widening the framework's adoption and making it more accessible in different machine learning ecosystems. πŸš€

@glenn-jocher glenn-jocher self-assigned this May 22, 2022
@glenn-jocher glenn-jocher merged commit 547c89b into master May 22, 2022
@glenn-jocher glenn-jocher deleted the glenn-jocher-patch-1 branch May 22, 2022 11:33
tdhooghe pushed a commit to tdhooghe/yolov5 that referenced this pull request Jun 10, 2022
ctjanuhowski pushed a commit to ctjanuhowski/yolov5 that referenced this pull request Sep 8, 2022
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.

Converting the Tensorflow saved_model in to a keras model '_UserObject' object has no attribute 'summary'
1 participant