File size: 356 Bytes
c3e16bb | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | from torchvision.transforms import Normalize, Compose, Resize, ToTensor
def convert_to_rgb(image):
return image.convert("RGB")
def get_transform(image_size=384):
return Compose([
convert_to_rgb,
Resize((image_size, image_size)),
ToTensor(),
Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
])
|