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

Improved dataset error introspection #8091

Merged
merged 2 commits into from
Jun 3, 2022
Merged

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Jun 3, 2022

Intended to help #8090

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhanced dataset validation and automatic configuration in the YOLOv5 framework.

πŸ“Š Key Changes

  • Combined dataset downloading, checking, and unzipping into a single function for simplicity.
  • Added checks within the check_dataset function to ensure provided data has necessary keys like 'train', 'val', 'nc' (number of classes), and assigns default 'names' if missing.
  • Removed redundant assertions and name assignments outside the initial data checks to streamline the code.

🎯 Purpose & Impact

  • The changes improve user experience by performing more comprehensive and upfront checks on datasets, which helps prevent common errors during training.
  • Default class names are now automatically generated if not provided, simplifying dataset setup for new users.
  • These improvements result in a more robust and user-friendly setup process when preparing data for model training, enhancing overall accessibility for both new and experienced users of YOLOv5. πŸš€

@glenn-jocher glenn-jocher self-assigned this Jun 3, 2022
@glenn-jocher glenn-jocher merged commit 47233e1 into master Jun 3, 2022
@glenn-jocher glenn-jocher deleted the update/data_checks branch June 3, 2022 10:45
tdhooghe pushed a commit to tdhooghe/yolov5 that referenced this pull request Jun 10, 2022
* Improved dataset error introspection

Intended to help ultralytics#8090

* Update general.py
ctjanuhowski pushed a commit to ctjanuhowski/yolov5 that referenced this pull request Sep 8, 2022
* Improved dataset error introspection

Intended to help ultralytics#8090

* Update general.py
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