(crop_size: int = CROP_SIZE)
| 85 | # ============================================================================ |
| 86 | |
| 87 | def make_train_transform(crop_size: int = CROP_SIZE) -> transforms.Compose: |
| 88 | return transforms.Compose([ |
| 89 | transforms.RandomResizedCrop(crop_size, interpolation=transforms.InterpolationMode.BICUBIC), |
| 90 | transforms.RandomHorizontalFlip(), |
| 91 | transforms.ToTensor(), |
| 92 | transforms.Normalize(mean=IMAGENET_DEFAULT_MEAN, std=IMAGENET_DEFAULT_STD), |
| 93 | ]) |
| 94 | |
| 95 | |
| 96 | def make_eval_transform(resize_size: int = RESIZE_SIZE, crop_size: int = CROP_SIZE) -> transforms.Compose: |
no outgoing calls
no test coverage detected