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

Update albumentations #9503

Merged
merged 11 commits into from
Sep 21, 2022
Merged

Update albumentations #9503

merged 11 commits into from
Sep 21, 2022

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Sep 19, 2022

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhancements in Albumentations data augmentation for YOLOv5.

πŸ“Š Key Changes

  • πŸ” Added size parameter to Albumentations class constructor and to the classify_albumentations function for dynamic resizing.
  • πŸ–Ό Introduced ratio parameter in classify_albumentations to control aspect ratio during random resized crops.
  • πŸ”„ The Albumentations class now passes img_size to the class constructor when called in dataloaders.

🎯 Purpose & Impact

  • πŸ“ The new size parameter allows custom sizing for image augmentation, improving flexibility for different datasets and model requirements.
  • πŸ“ The ratio parameter provides additional control over the aspect ratio of images, which can lead to better performance and more robust models.
  • 🌱 These changes can potentially improve YOLOv5's performance by allowing more precise image preprocessing that is better suited to the target use case.

@glenn-jocher glenn-jocher self-assigned this Sep 19, 2022
pre-commit-ci bot and others added 5 commits September 19, 2022 23:14
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
@glenn-jocher glenn-jocher merged commit 6f02847 into master Sep 21, 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