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

Single class train update #1719

Merged
merged 1 commit into from Dec 17, 2020
Merged

Single class train update #1719

merged 1 commit into from Dec 17, 2020

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Dec 17, 2020

Fix for 'item' class appearing on forced --single-cls training of existing single class datasets. See #12 (comment)

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Refinement of single-class mode training in YOLOv5.

πŸ“Š Key Changes

  • Tweaked the logic setting up the number of classes (nc) and the names of the classes (names) for single-class datasets in train.py.
  • Amended the help description for the --single-cls argument to clarify its purpose.

🎯 Purpose & Impact

  • 🎯 The adjustment ensures that single-class training mode correctly handles the class names within the dataset, whether there is only one name or multiple names present.
  • πŸš€ This update improves usability by providing clear guidance when training YOLOv5 on single-class datasets, potentially preventing errors and confusion for users.
  • πŸ“ˆ Users opting to train models with a single class can expect more robust behavior, especially when the dataset might inaccurately have multiple names listed.

@glenn-jocher glenn-jocher merged commit 7e161d9 into master Dec 17, 2020
@glenn-jocher glenn-jocher deleted the single_cls branch December 17, 2020 20:02
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.

None yet

1 participant