Skip to content

Instantly share code, notes, and snippets.

@souravs17031999

souravs17031999/cifar10transform.py

Last active July 21, 2019 12:37
Show Gist options
  • Save souravs17031999/5e47fe2d9f3f4b3e7261c5a251449d3b to your computer and use it in GitHub Desktop.
Save souravs17031999/5e47fe2d9f3f4b3e7261c5a251449d3b to your computer and use it in GitHub Desktop.
Download ZIP
Raw
cifar10transform.py
train_transform = transforms.Compose([transforms.RandomRotation(30),
transforms.RandomResizedCrop(224),
transforms.RandomHorizontalFlip(),
transforms.ColorJitter(brightness=0.4, contrast=0.4, saturation=0.4, hue=0.1),
transforms.ToTensor(),
transforms.Normalize([0.5,],[0.5,])])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment