Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Update augmentations.py (#3948)

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
utils/augmentations.py
+1 -1
View File
@@ -25,7 +25,7 @@ class Albumentations:
A.ToGray(p=0.01)], A.ToGray(p=0.01)],
bbox_params=A.BboxParams(format='yolo', label_fields=['class_labels'])) 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 except ImportError: # package not installed, skip
pass pass
except Exception as e: except Exception as e: