Python Data Augmentation Tensorflow Tutorial Medium