Gan Data Augmentation Python