| from torchvision.transforms import Normalize, Compose, Resize, ToTensor | |
| def get_transform(image_size=384): | |
| return Compose([ | |
| lambda image: image.convert("RGB"), | |
| Resize((image_size, image_size)), | |
| ToTensor(), | |
| Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]) | |
| ]) | |