Duplicated from Sher1988/image_classifier
38e36bb
1
2
3
from torchvision.models import ResNet50_Weights transforms = ResNet50_Weights.DEFAULT.transforms()