glenn-jocher commited on
Commit
7933832
1 Parent(s): d856c48

Update default Albumentations (#4931)

Browse files
Files changed (1) hide show
  1. utils/augmentations.py +7 -3
utils/augmentations.py CHANGED
@@ -23,9 +23,13 @@ class Albumentations:
23
  check_version(A.__version__, '1.0.3') # version requirement
24
 
25
  self.transform = A.Compose([
26
- A.Blur(p=0.1),
27
- A.MedianBlur(p=0.1),
28
- A.ToGray(p=0.01)],
 
 
 
 
29
  bbox_params=A.BboxParams(format='yolo', label_fields=['class_labels']))
30
 
31
  logging.info(colorstr('albumentations: ') + ', '.join(f'{x}' for x in self.transform.transforms if x.p))
 
23
  check_version(A.__version__, '1.0.3') # version requirement
24
 
25
  self.transform = A.Compose([
26
+ A.Blur(p=0.01),
27
+ A.MedianBlur(p=0.01),
28
+ A.ToGray(p=0.01),
29
+ A.CLAHE(p=0.01),
30
+ A.RandomBrightnessContrast(p=0.0),
31
+ A.RandomGamma(p=0.0),
32
+ A.ImageCompression(quality_lower=75, p=0.0)],
33
  bbox_params=A.BboxParams(format='yolo', label_fields=['class_labels']))
34
 
35
  logging.info(colorstr('albumentations: ') + ', '.join(f'{x}' for x in self.transform.transforms if x.p))