Skip to content

Commit a26e7de

Browse files
authored
Update augmentations.py (#3948)
1 parent 248504c commit a26e7de

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

utils/augmentations.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ def __init__(self):
2525
A.ToGray(p=0.01)],
2626
bbox_params=A.BboxParams(format='yolo', label_fields=['class_labels']))
2727

28-
logging.info(colorstr('albumentations: ') + ', '.join(f'{x}' for x in self.transform.transforms))
28+
logging.info(colorstr('albumentations: ') + ', '.join(f'{x}' for x in self.transform.transforms if x.p))
2929
except ImportError: # package not installed, skip
3030
pass
3131
except Exception as e:

0 commit comments

Comments
 (0)