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: