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

auto-verbose if nc <= 20 #1869

Merged
merged 1 commit into from
Jan 7, 2021
Merged

auto-verbose if nc <= 20 #1869

merged 1 commit into from
Jan 7, 2021

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Jan 7, 2021

This PR automatically sets test.py to --verbose for datasets with <= 20 classes when called directly (not training).

This may help most users see per-class metrics by default since typically custom datasets are < 20 classes.

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhanced information displayed for per-class results during testing.

πŸ“Š Key Changes

  • Modified the condition that determines when to print per-class results in test.py.
  • Per-class results will now print either when verbose is true, or when the number of classes (nc) is less than or equal to 20 and the model is not in training mode.

🎯 Purpose & Impact

  • πŸ’‘ Purpose: To improve the usability by providing more detailed feedback on model performance when the number of classes is small, without overwhelming the user with information during training sessions.
  • πŸš€ Potential Impact: Users can now get a quick overview of model accuracy per class for small datasets even when not in verbose mode, which is great for quick evaluations and debugging. This could be particularly useful when working with a limited number of classes and when users want an immediate, detailed assessment without sifting through extensive logs.

@glenn-jocher glenn-jocher linked an issue Jan 7, 2021 that may be closed by this pull request
@glenn-jocher glenn-jocher changed the title auto-verbose if nc <=20 auto-verbose if nc <= 20 Jan 7, 2021
@glenn-jocher glenn-jocher merged commit 3e25f1e into master Jan 7, 2021
@glenn-jocher glenn-jocher deleted the glenn-jocher-patch-1 branch January 7, 2021 23:47
KMint1819 pushed a commit to KMint1819/yolov5 that referenced this pull request May 12, 2021
taicaile pushed a commit to taicaile/yolov5 that referenced this pull request Oct 12, 2021
BjarneKuehl pushed a commit to fhkiel-mlaip/yolov5 that referenced this pull request Aug 26, 2022
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.

How to output test.py metrics per class?
1 participant