def __init__(
            self,
            # min_sizes: List[int] = (416, 512, 608, 704, 800, 1280, 1408, 1536, 1664),
            # max_size: List[int] = (1280, 1280, 1280, 1280, 1280, 1280, 1408, 1536, 1664),
            min_sizes: List[int] = None,
            max_size: List[int] = None,
            flip: bool = False,
    ):
        """
        Default args are from d2 Mask RCNN

        Args:
            min_sizes: list of short-edge size to resize the image to
            max_size: maximum height or width of resized images
            flip: whether to apply flipping augmentation
        """
        assert len(min_sizes) == len(max_size)
        self.min_sizes = min_sizes
        self.max_size = max_size
        self.flip = flip

        Default args are from d2 Mask RCNN

        Args:
            min_sizes: list of short-edge size to resize the image to
            max_size: maximum height or width of resized images
            flip: whether to apply flipping augmentation
        
EVA-02/det/detectron2/modeling/test_time_augmentation.py