Python Code For Data Augmentation Tensorflow