(resize_size: int = RESIZE_SIZE, crop_size: int = CROP_SIZE)
| 94 | |
| 95 | |
| 96 | def make_eval_transform(resize_size: int = RESIZE_SIZE, crop_size: int = CROP_SIZE) -> transforms.Compose: |
| 97 | return transforms.Compose([ |
| 98 | transforms.Resize(resize_size, interpolation=transforms.InterpolationMode.BICUBIC), |
| 99 | transforms.CenterCrop(crop_size), |
| 100 | transforms.ToTensor(), |
| 101 | transforms.Normalize(mean=IMAGENET_DEFAULT_MEAN, std=IMAGENET_DEFAULT_STD), |
| 102 | ]) |
| 103 | |
| 104 | |
| 105 | # ============================================================================ |
no outgoing calls
no test coverage detected