Pytorch Image Augmentation For Deep