Skip to content

Instantly share code, notes, and snippets.

Avatar
:octocat:
Busy

Author Morgan Auth0rM0rgan

:octocat:
Busy
  • @own_business
  • FindMe
View GitHub Profile
View tf_data_augmentation_on_gpu.py
def augment(images, labels,
resize=None, # (width, height) tuple or None
horizontal_flip=False,
vertical_flip=False,
rotate=0, # Maximum rotation angle in degrees
crop_probability=0, # How often we do crops
crop_min_percent=0.6, # Minimum linear dimension of a crop
crop_max_percent=1., # Maximum linear dimension of a crop
mixup=0): # Mixup coeffecient, see https://arxiv.org/abs/1710.09412.pdf
if resize is not None: