| 15 | |
| 16 | |
| 17 | class ImagePreprocessor: |
| 18 | def __init__(self, resolution: Tuple[int, int] = (1024, 1024)) -> None: |
| 19 | self.transform_image = transforms.Compose( |
| 20 | [ |
| 21 | transforms.Resize(resolution), |
| 22 | transforms.ToTensor(), |
| 23 | transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]), |
| 24 | ] |
| 25 | ) |
| 26 | |
| 27 | def proc(self, image: Image.Image) -> torch.Tensor: |
| 28 | image = self.transform_image(image) |
| 29 | return image |
| 30 | |
| 31 | |
| 32 | usage_to_weights_file = { |