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

Simplify dataloader tqdm descriptions #10210

Merged
merged 3 commits into from
Nov 18, 2022
Merged

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Nov 18, 2022

@AyushExel this should help our tqdm dataloader messages fit better within a single line in our Colab notebooks (problem shown below) and also help avoid confusion about missing/empty labels, now combined into 'backgrounds'.

Screenshot 2022-11-18 at 20 50 47

Signed-off-by: Glenn Jocher [email protected]

🛠️ PR Summary

Made with ❤️ by Ultralytics Actions

🌟 Summary

Improvements to dataset scanning and labeling output messages in the Ultralytics YOLOv5 project.

📊 Key Changes

  • Removed colorstr, check_dataset, check_requirements, and check_yaml imports, which were no longer needed directly in dataloaders.py.
  • Refined printed output messages for dataset scanning to provide clearer statistics on image datasets (e.g., how many images were found, how many backgrounds, and the number of corrupt files).

🎯 Purpose & Impact

  • 🧹 Cleanup: The removal of unnecessary imports declutters the code and may marginally improve the performance.
  • 📈 Clarity & UX: The updated messages during dataset scanning give users more intuitive feedback on their datasets. This helps in troubleshooting issues related to missing or corrupt files more easily.
  • 📊 Statistics: Combining 'missing' and 'empty' labels into 'backgrounds' provides a more compact and understandable summary for users to quickly assess the state of their datasets.

@AyushExel this should help our tqdm dataloader messages fit better within a single line in our Colab notebooks and also help avoid confusion about missing/empty labels, now combined into 'backgrounds'.

Signed-off-by: Glenn Jocher <[email protected]>
@glenn-jocher glenn-jocher self-assigned this Nov 18, 2022
@glenn-jocher glenn-jocher merged commit 74b3886 into master Nov 18, 2022
@glenn-jocher glenn-jocher deleted the glenn-jocher-patch-3 branch November 18, 2022 20:18
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