glenn-jocher
commited on
Commit
•
7933832
1
Parent(s):
d856c48
Update default Albumentations (#4931)
Browse files- 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.
|
27 |
-
A.MedianBlur(p=0.
|
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))
|