Image Augmentation Using Pytorch