Using data augmentation in deep learning libraries