Modelpublic

ZhengPeng7/BiRefNet-COD

sourceHugging Facemitupdated 25d agoView on Hugging Face

1likes849downloads

BiRefNet_config.py12 linesDownload Raw Back to root

1from transformers import PretrainedConfig2 3class BiRefNetConfig(PretrainedConfig):4    model_type = "SegformerForSemanticSegmentation"5    def __init__(6        self,7        bb_pretrained=False,8        **kwargs9    ):10        self.bb_pretrained = bb_pretrained11        super().__init__(**kwargs)12