Update augmentations.py (#3948)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-07-09 16:45:04 +02:00
1 parent 248504cf13
commit a26e7de2bf
1 file changed
+1 -1
+1 -1
View File
@@ -25,7 +25,7 @@ class Albumentations:
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: