Danny_Net_Demo / preprocessing.py
cfgpp's picture
Upload 2 files
2885278 verified
raw
history blame contribute delete
332 Bytes
# 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)