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

val.py assert ncm == nc fix #8545

Merged
merged 1 commit into from
Jul 11, 2022
Merged

val.py assert ncm == nc fix #8545

merged 1 commit into from
Jul 11, 2022

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Jul 11, 2022

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhanced error message for model and dataset class mismatch in validation script.

πŸ“Š Key Changes

  • Modified the assertion error message in val.py to display the complete weights array when a class mismatch is detected.

🎯 Purpose & Impact

  • Purpose: To provide a more informative error message to users when their model's number of classes doesn't match the dataset classes during validation.
  • Impact: This change will help users quickly identify issues when using datasets with a different number of classes than the model was trained on, improving debugging and user experience. βš™οΈπŸ”

@glenn-jocher glenn-jocher self-assigned this Jul 11, 2022
@glenn-jocher glenn-jocher merged commit 7dafd1c into master Jul 11, 2022
@glenn-jocher glenn-jocher deleted the glenn-jocher-patch-1 branch July 11, 2022 13:09
Shivvrat pushed a commit to Shivvrat/epic-yolov5 that referenced this pull request Jul 12, 2022
ctjanuhowski pushed a commit to ctjanuhowski/yolov5 that referenced this pull request Sep 8, 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.

1 participant