zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Update default Albumentations (#4931)

This commit is contained in:
Glenn Jocher 2021-09-26 12:10:33 -07:00 committed by GitHub
parent d856c48298
commit 793383232f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
utils/augmentations.py
View File

@ -23,9 +23,13 @@ class Albumentations:
check_version(A.__version__, '1.0.3') # version requirement check_version(A.__version__, '1.0.3') # version requirement
self.transform = A.Compose([ self.transform = A.Compose([
A.Blur(p=0.1), A.Blur(p=0.01),
A.MedianBlur(p=0.1), A.MedianBlur(p=0.01),
A.ToGray(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'])) 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)) logging.info(colorstr('albumentations: ') + ', '.join(f'{x}' for x in self.transform.transforms if x.p))