def __init__(
        self,
        is_train=True,
        size_train=None,
        size_test=None,
        size_crop=None,
    ):
        """
        NOTE: this interface is experimental.
        Args:
            is_train: for training or inference
            augmentations: a list of augmentations or deterministic transforms to apply
            tfm_gens: data augmentation
            image_format: an image format supported by :func:`detection_utils.read_image`.
        """
        self.is_train = is_train
        self.size_train = size_train
        self.size_test = size_test
        self.size_crop = size_crop

        t = []
        t.append(transforms.Resize(size_crop, interpolation=Image.BICUBIC))
        t.append(transforms.CenterCrop(size_test))
        self.transform = transforms.Compose(t)

        NOTE: this interface is experimental.
        Args:
            is_train: for training or inference
            augmentations: a list of augmentations or deterministic transforms to apply
            tfm_gens: data augmentation
            image_format: an image format supported by :func:`detection_utils.read_image`.
        
datasets/dataset_mappers/imagenet_dataset_mapper.py