Spaces:
Running
Running
# 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) | |