Commit
•
a26e7de
1
Parent(s):
248504c
Update augmentations.py (#3948)
Browse files- utils/augmentations.py +1 -1
utils/augmentations.py
CHANGED
|
@@ -25,7 +25,7 @@ class Albumentations:
|
|
| 25 |
A.ToGray(p=0.01)],
|
| 26 |
bbox_params=A.BboxParams(format='yolo', label_fields=['class_labels']))
|
| 27 |
|
| 28 |
-
logging.info(colorstr('albumentations: ') + ', '.join(f'{x}' for x in self.transform.transforms))
|
| 29 |
except ImportError: # package not installed, skip
|
| 30 |
pass
|
| 31 |
except Exception as e:
|
|
|
|
| 25 |
A.ToGray(p=0.01)],
|
| 26 |
bbox_params=A.BboxParams(format='yolo', label_fields=['class_labels']))
|
| 27 |
|
| 28 |
+
logging.info(colorstr('albumentations: ') + ', '.join(f'{x}' for x in self.transform.transforms if x.p))
|
| 29 |
except ImportError: # package not installed, skip
|
| 30 |
pass
|
| 31 |
except Exception as e:
|