import torch from torchvision import transforms transforms_img = transforms.Compose([ transforms.Resize((224, 224)), # 1. Reshape all images to 224x224 (though some models may require different sizes) transforms.ToTensor(), # 2. Turn image values to between 0 & 1 transforms.Normalize(mean=[0.1909, 0.1937, 0.1896], std=[0.3242, 0.3258, 0.3336]) ])