From a26e7de2bffaf5a87e7ed83aeabd0f0b2e8ad861 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 9 Jul 2021 16:45:04 +0200 Subject: [PATCH] Update augmentations.py (#3948) --- utils/augmentations.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/augmentations.py b/utils/augmentations.py index 74ee4de2131e..81652c191bc1 100644 --- a/utils/augmentations.py +++ b/utils/augmentations.py @@ -25,7 +25,7 @@ def __init__(self): A.ToGray(p=0.01)], bbox_params=A.BboxParams(format='yolo', label_fields=['class_labels'])) - logging.info(colorstr('albumentations: ') + ', '.join(f'{x}' for x in self.transform.transforms)) + logging.info(colorstr('albumentations: ') + ', '.join(f'{x}' for x in self.transform.transforms if x.p)) except ImportError: # package not installed, skip pass except Exception as e: