def __init__(self, size: Union[int, List[int], Tuple[int]], p: float = 0.5) -> None:
        """Randomly Crops the image.

        Args:
            output_size: height and width of the crop box. If int, this size is used for both directions.
        """
        self.size = (size, size) if isinstance(size, int) else size
        self.p = p
Randomly Crops the image.

        Args:
            output_size: height and width of the crop box. If int, this size is used for both directions.
        
semseg/augmentations.py