Spaces:
Sleeping
Sleeping
| 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]) | |
| ]) | |