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

Fix missing imports #627

Merged
merged 2 commits into from
Aug 4, 2020

Conversation

lorenzomammana
Copy link
Contributor

@lorenzomammana lorenzomammana commented Aug 4, 2020

Latest changes have broken certain imports in train.py and detect.py, i'm using this PR as a fix.

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhancements in YOLOv5 detection and training scripts for better utility.

πŸ“Š Key Changes

  • In detect.py, the strip_optimizer function from utils.general has been made available.
  • In train.py, the plot_evolution function has been added to the imports from utils.general.

🎯 Purpose & Impact

  • Importing the strip_optimizer in detect.py could be preparing the script to allow users to strip optimizers from models for easier deployment or inference-time optimizations. πŸš€
  • Adding plot_evolution in train.py suggests new functionality related to visualizing the evolution or progress of certain metrics during the training process, which can provide users with better insights into model performance over time. πŸ“ˆ
  • Overall, these changes will likely improve the utility and user experience for those working with the YOLOv5 model, by providing additional tools for model optimization and training analysis. πŸ› οΈ

@glenn-jocher
Copy link
Member

@lorenzomammana thank you, good catch!

@glenn-jocher glenn-jocher merged commit 728efa6 into ultralytics:master Aug 4, 2020
@lorenzomammana lorenzomammana deleted the fix-missing-imports branch August 6, 2020 07:21
burglarhobbit pushed a commit to burglarhobbit/yolov5 that referenced this pull request Jan 1, 2021
* Fix missing imports

* Update detect.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
KMint1819 pushed a commit to KMint1819/yolov5 that referenced this pull request May 12, 2021
* Fix missing imports

* Update detect.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
BjarneKuehl pushed a commit to fhkiel-mlaip/yolov5 that referenced this pull request Aug 26, 2022
* Fix missing imports

* Update detect.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
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