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 --noplots flag to suppress figures and images logging #7534

Merged
merged 12 commits into from
Apr 22, 2022

Conversation

AyushExel
Copy link
Contributor

@AyushExel AyushExel commented Apr 22, 2022

Addresses -> #7484
This PR adds support for a argparse flag nomedia which prevent any media logging locally or W&B dashboard.
W&B dash with nomedia enabled -> https://wandb.ai/acwb/YOLOv5/runs/4shjupfd?workspace=user-
Screenshot 2022-04-22 at 6 08 54 PM
Local run folder
Screenshot 2022-04-22 at 6 09 12 PM

@glenn-jocher you might need to take a closer look as this touches some non-wandb code.

🛠️ PR Summary

Made with ❤️ by Ultralytics Actions

🌟 Summary

Enhanced training options for generating plots and synchronized batch normalization in YOLOv5.

📊 Key Changes

  • Added --noplots flag to disable plot generation during training.
  • Removed sync_bn parameter from on_train_batch_end callback and accessed it directly from the options.
  • Adjusted conditions for setting bbox_interval to also consider the new --noplots flag.

🎯 Purpose & Impact

  • Disabling Plots: Users can now skip the creation of plot files during training, which can save time and resources, especially when plots are not needed.
  • Simplified Synchronization: Streamlining the code by removing an extra parameter and preventing known issues with torch's tb.add_graph() when using synchronized batch normalization.
  • Bbox Interval Adjustment: Ensures that bounding box intervals are configured correctly in relation to plot generation preferences, providing more control over when evaluation metrics are logged during training.

With these changes, users gain more flexibility in how they train their models, potentially speeding up the training process and focusing system resources on essential tasks. 🚀

@glenn-jocher glenn-jocher changed the title Add support for nomedia logging for local and W&B Add --noplots flag to suppress figures/image logging Apr 22, 2022
@glenn-jocher glenn-jocher changed the title Add --noplots flag to suppress figures/image logging Add --noplots flag to suppress figures and images logging Apr 22, 2022
@glenn-jocher glenn-jocher merged commit 4cac1cf into ultralytics:master Apr 22, 2022
@glenn-jocher glenn-jocher linked an issue Apr 22, 2022 that may be closed by this pull request
BjarneKuehl pushed a commit to fhkiel-mlaip/yolov5 that referenced this pull request Aug 26, 2022
…ics#7534)

* support nomedia

* support nomedia for validation

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

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

* Update train.py

* Revert no plot evolve

evolve plots do not contain any images

* Revert plot_results

contains no media

* Update wandb_utils.py

* sync-bn cleanup

* Cleanup

* Rename nomedia -> noplots

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
ctjanuhowski pushed a commit to ctjanuhowski/yolov5 that referenced this pull request Sep 8, 2022
…ics#7534)

* support nomedia

* support nomedia for validation

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

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

* Update train.py

* Revert no plot evolve

evolve plots do not contain any images

* Revert plot_results

contains no media

* Update wandb_utils.py

* sync-bn cleanup

* Cleanup

* Rename nomedia -> noplots

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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.

Wandb logging without storing images
2 participants