Unverified 提交 79338323 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Update default Albumentations (#4931)

上级 d856c482
...@@ -23,9 +23,13 @@ class Albumentations: ...@@ -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))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论