Skip to content

Commit

Permalink
add spaces
Browse files Browse the repository at this point in the history
  • Loading branch information
glenn-jocher committed Jul 5, 2021
1 parent 371991a commit 4139fec
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions utils/augmentations.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,11 +18,13 @@ def __init__(self):
try:
import albumentations as A
check_version(A.__version__, '1.0.0') # version requirement

self.transform = A.Compose([
A.Blur(p=0.1),
A.MedianBlur(p=0.1),
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))
except ImportError: # package not installed, skip
pass
Expand Down

0 comments on commit 4139fec

Please sign in to comment.