| """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 | |