How to export the model as keras h5 file？

[lebron8dong]

Search before asking

• I have searched the YOLOv5 issues and found no similar bug report.

YOLOv5 Component

No response

Bug

"""
TensorFlow, Keras and TFLite versions of YOLOv5
Authored by https://github.com/zldrobit in PR #1127

Usage:
$ python models/tf.py --weights yolov5s.pt

Export:
$ python export.py --weights yolov5s.pt --include saved_model pb tflite tfjs
"""

There is no h5 file after I execute the command

Environment

No response

Minimal Reproducible Example

No response

Additional

No response

Are you willing to submit a PR?

• Yes I'd like to help by submitting a PR!

[github-actions]

👋 Hello there! We wanted to give you a friendly reminder that this issue has not had any recent activity and may be closed soon, but don't worry - you can always reopen it if needed. If you still have any questions or concerns, please feel free to let us know how we can help.

For additional resources and information, please see the links below:

• Docs: https://docs.ultralytics.com
• HUB: https://hub.ultralytics.com
• Community: https://community.ultralytics.com

Feel free to inform us of any other issues you discover or feature requests that come to mind in the future. Pull Requests (PRs) are also always welcomed!

Thank you for your contributions to YOLO 🚀 and Vision AI ⭐

[glenn-jocher]

@lebron8dong to export the YOLOv5 model as a Keras h5 file, you can use the provided export script with the --include keras option. After executing the command, the h5 file should be generated accordingly.

For more details on model export, refer to the Ultralytics Docs at https://docs.ultralytics.com/yolov5/.

Let me know if you need further assistance!