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 default Albumentations #4931

Merged
merged 1 commit into from
Sep 26, 2021
Merged

Update default Albumentations #4931

merged 1 commit into from
Sep 26, 2021

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Sep 26, 2021

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Optimized image augmentation probabilities in the YOLOv5 model.

πŸ“Š Key Changes

  • Reduced the probability of applying the Blur and MedianBlur effects from 0.1 to 0.01.
  • Added new augmentations with a probability of 0.01 for CLAHE (Contrast Limited Adaptive Histogram Equalization).
  • Introduced RandomBrightnessContrast and RandomGamma augmentations with a disabled state (probability set to 0.0).
  • Included ImageCompression with a default quality setting of 75 but disabled it (probability set to 0.0).

🎯 Purpose & Impact

  • 🎨 Enhanced Data Augmentation: Adding new transformations like CLAHE enriches the data augmentation pipeline and can potentially improve model robustness to various lighting conditions.
  • πŸ” Fine-Tuning Probabilities: Adjusting the probabilities of existing augmentations aims to create a more balanced augmentation strategy that may lead to better generalization and prevent overfitting.
  • πŸš€ Experimentation Ready: Including new augmentations with a zero probability allows for quick experimentation. Developers can easily enable these augmentations in the future for further research and testing.
  • πŸ“ˆ Potential Impact: These changes could lead to more effective training and possibly improved accuracy without changing the current model behavior since new augmentations are initially set to not apply.

@glenn-jocher glenn-jocher merged commit 7933832 into master Sep 26, 2021
@glenn-jocher glenn-jocher deleted the glenn-jocher-patch-1 branch September 26, 2021 19:10
CesarBazanAV pushed a commit to CesarBazanAV/yolov5 that referenced this pull request Sep 29, 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