Update default Albumentations (#4931)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-09-26 12:10:33 -07:00
1 parent d856c48298
commit 793383232f
1 file changed
+7 -3
+7 -3
View File
@@ -23,9 +23,13 @@ class Albumentations:
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))