Python Data Augmentation Tensorflow