Skip to content

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.

1 participant