from torchvision.models import ResNet50_Weights

transforms = ResNet50_Weights.DEFAULT.transforms()