Skip to content

Instantly share code, notes, and snippets.

@J3698

J3698/transforms.py

Last active June 27, 2021 17:57
Show Gist options
  • Save J3698/36065f207c15625b5680d03be21fead3 to your computer and use it in GitHub Desktop.
Save J3698/36065f207c15625b5680d03be21fead3 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
transforms.py
def get_transforms(crop):
if crop:
return Compose([Resize(512), RandomCrop(256), ToTensor()])
return Compose([Resize((256, 256)), ToTensor()])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment