@@ -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: |