Auth0rM0rgan
/ tf_data_augmentation_on_gpu.py
Created Apr 17, 2018
— forked from vertix/tf_data_augmentation_on_gpu.py
TF data augmentation on GPU
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: |