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

W&B: refactor W&B tables #5737

Merged
merged 9 commits into from
Nov 25, 2021
Merged

W&B: refactor W&B tables #5737

merged 9 commits into from
Nov 25, 2021

Conversation

AyushExel
Copy link
Contributor

@AyushExel AyushExel commented Nov 21, 2021

This PR:

  • refactors table integration to include avg. conf. of each class detected.
  • Allows the user to upload only validation split of the dataset
  • Improves W&B readme

Before:

Screenshot from 2021-11-21 17-35-12

After:

users can now sort/filter using any class, which enables inspecting on which classes models performs best/worst

Screenshot from 2021-11-21 17-40-06

TODO:

  • Format using Pycharm
  • Run additional tests

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhancements to Weights & Biases logging in YOLOv5 training script.

πŸ“Š Key Changes

  • Updated the --upload_dataset argument to include an optional "val" value, allowing for more flexible dataset uploads.
  • Added documentation on how to disable and enable Weights & Biases (wandb) logging.
  • Introduced efficiency improvements by avoiding repeat image uploads when using evaluation tables.
  • Adjusted the wandb logger to log average confidence per class during training rather than total confidence.
  • Several internal code refinements within wandb_utils.py for better readability and maintainability.

🎯 Purpose & Impact

  • The changes provide users with more control over dataset uploads to wandb and an option to disable logging if desired. πŸŽ›οΈ
  • Enhanced evaluation tables now provide more detailed insights into model performance, potentially aiding in debugging and model improvement. πŸ“ˆ
  • Streamlined usage and clarified documentation make it easier for both new and existing users to utilize wandb features with YOLOv5. πŸ“š
  • Overall, these updates aim to sharpen the development experience and provide richer datasets for those integrating YOLOv5 with wandb. πŸ”

@glenn-jocher
Copy link
Member

@AyushExel thanks for the PR! Please fix pre-commit errors here:
https://results.pre-commit.ci/run/github/264818686/1637496775.8WDi9EjUQA6h8x5dKIZVfQ

@AyushExel AyushExel changed the title [WIP] W&B: refactor W&B tables W&B: refactor W&B tables Nov 25, 2021
@AyushExel AyushExel marked this pull request as ready for review November 25, 2021 09:54
@AyushExel
Copy link
Contributor Author

@glenn-jocher I've updated the README.md in wandb folder with a section on how to enable/disable. I've also made some updates to reflect the current updates.
Can you please copy the entire markdown form that files to the official Wandb Tutorial? This PR is ready to be merged

@glenn-jocher
Copy link
Member

@AyushExel great! Can you provide the links of exactly what to copy to what? Thanks!

@AyushExel
Copy link
Contributor Author

@glenn-jocher after merging this PR, copy the contents from this wand Readme.md to the yolov5 tutorial issue

@glenn-jocher glenn-jocher merged commit f2ca30a into ultralytics:master Nov 25, 2021
@glenn-jocher
Copy link
Member

@AyushExel got it! PR is merged and https://github.com/ultralytics/yolov5/issues/1289 is updated.

Thank you for your contributions to YOLOv5 πŸš€ and Vision AI ⭐

BjarneKuehl pushed a commit to fhkiel-mlaip/yolov5 that referenced this pull request Aug 26, 2022
* update

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

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

* reformat

* Single-line argparser argument

* Update README.md

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

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

* Update README.md

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

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

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.

None yet

2 participants