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 ConfusionMatrix normalize=True flag #3586

Merged
merged 1 commit into from
Jun 11, 2021
Merged

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Jun 11, 2021

Per #3533

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Improved customizable confusion matrix plotting in YOLOv5.

πŸ“Š Key Changes

  • Added normalize parameter to the plot method within utils/metrics.py.
  • Enabled optional normalization of the confusion matrix before plotting.

🎯 Purpose & Impact

  • The normalize parameter allows users to choose whether or not they want to see their confusion matrix normalized, which provides flexibility in analyzing model performance.
  • This can be especially useful for users wanting to understand the raw number of cases for each prediction category, or compare different models' performance more directly.
  • The impact is mainly on improving the usability of the model evaluation tools, potentially making it easier for developers and data scientists to tune their models and communicate results. πŸ“ˆ

@glenn-jocher glenn-jocher linked an issue Jun 11, 2021 that may be closed by this pull request
@glenn-jocher glenn-jocher merged commit ec2da4a into master Jun 11, 2021
@glenn-jocher glenn-jocher deleted the glenn-jocher-patch-4 branch June 11, 2021 09:37
Lechtr pushed a commit to Lechtr/yolov5 that referenced this pull request Jul 20, 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 know our accuracy
1 participant