Python Data Augmentation Tensorflow Hub