Update default Albumentations (ultralytics#4931)

fhkiel-mlaip · Sep 26, 2021 · 28905cc · 28905cc
1 parent fb7ad78
commit 28905cc
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/utils/augmentations.py b/utils/augmentations.py
@@ -23,9 +23,13 @@ def __init__(self):
             check_version(A.__version__, '1.0.3')  # version requirement
 
             self.transform = A.Compose([
-                A.Blur(p=0.1),
-                A.MedianBlur(p=0.1),
-                A.ToGray(p=0.01)],
+                A.Blur(p=0.01),
+                A.MedianBlur(p=0.01),
+                A.ToGray(p=0.01),
+                A.CLAHE(p=0.01),
+                A.RandomBrightnessContrast(p=0.0),
+                A.RandomGamma(p=0.0),
+                A.ImageCompression(quality_lower=75, p=0.0)],
                 bbox_params=A.BboxParams(format='yolo', label_fields=['class_labels']))
 
             logging.info(colorstr('albumentations: ') + ', '.join(f'{x}' for x in self.transform.transforms if x.p))