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

Daemon plot_labels() for faster start #9057

Merged
merged 4 commits into from
Aug 21, 2022
Merged

Daemon plot_labels() for faster start #9057

merged 4 commits into from
Aug 21, 2022

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Aug 21, 2022

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhanced threading in callbacks and adjusted label plotting in training.

πŸ“Š Key Changes

  • Removed plot_labels call from train.py to streamline the training process.
  • Updated callbacks to potentially execute in daemon threads for asynchronous operation.
  • Simplified download_one comment in utils/general.py.
  • Integrated label plotting directly into a callback for cleaner management.
  • Minor adjustments to comments and code organization for clarity.

🎯 Purpose & Impact

  • πŸ‘ The removal of immediate label plotting during training reduces initial overhead and allows for a more focused start to the training process.
  • πŸš€ Allowing callbacks to run in separate threads can improve performance by not blocking the main thread, making training more efficient.
  • πŸ“ˆ The integration of label plotting into a callback could make it easier to manage custom logging or plotting routines.
  • 🧹 These changes lead to cleaner code structure and potentially better resource utilization during training, positively impacting users' experiences with YOLOv5 training sessions.

@glenn-jocher glenn-jocher self-assigned this Aug 21, 2022
@glenn-jocher glenn-jocher merged commit e0700cc into master Aug 21, 2022
@glenn-jocher glenn-jocher deleted the faster_start branch August 21, 2022 01:47
ctjanuhowski pushed a commit to ctjanuhowski/yolov5 that referenced this pull request Sep 8, 2022
* Daemon `plot_labels()` for faster start

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update train.py

Signed-off-by: Glenn Jocher <[email protected]>

Signed-off-by: Glenn Jocher <[email protected]>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.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

1 participant