from torchvision import transforms def get_transforms(): transform = transforms.Compose([ transforms.Resize((224, 224)), transforms.ToTensor(), transforms.Normalize( mean=[0.485, 0.456, 0.406], # ImageNet means std=[0.229, 0.224, 0.225] # ImageNet stds ) ]) return transform