Spaces:
Running
Running
File size: 332 Bytes
2885278 |
1 2 3 4 5 6 7 8 9 10 11 12 |
# utils/preprocessing.py
from torchvision import transforms
def preprocess_image(img):
transform = transforms.Compose([
transforms.Resize(256),
transforms.CenterCrop(224),
transforms.ToTensor(),
transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
])
return transform(img)
|