Update augmentations.py (ultralytics#3948)

TexasInstruments · Jul 9, 2021 · a26e7de · a26e7de
1 parent 248504c
commit a26e7de
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: