"""Configuration for SAC Patch Segmenter model."""

from transformers import PretrainedConfig


class SACPatchSegmenterConfig(PretrainedConfig):
    """Config for SAC U-Net patch segmenter that accepts PIL images directly.

    Reference: Liu et al., CVPR 2022, "Segment and Complete"
    """

    model_type = "sac-patch-segmenter"

    def __init__(self, base_filter: int = 16, input_size: int = 416, **kwargs):
        super().__init__(**kwargs)
        self.base_filter = base_filter
        self.input_size = input_size