Keras data augmentation with large dataset