Update augmentations.py (#3948)
This commit is contained in:
1 parent
248504cf13
commit
a26e7de2bf
1 file changed
+1
-1
@@ -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:
|
||||
|
||||
Reference in new issue
Block a user