Update augmentations.py (ultralytics#3948)

fhkiel-mlaip · Jul 9, 2021 · 356b4a1 · 356b4a1
1 parent a48064b
commit 356b4a1
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: