not-lain/CustomCodeForRMBG

1 2from transformers import PretrainedConfig3from typing import List4 5class RMBGConfig(PretrainedConfig):6    model_type = "SegformerForSemanticSegmentation"7    def __init__(8        self,9        in_ch=3,10        out_ch=1,11        **kwargs):12      self.in_ch = in_ch13      self.out_ch = out_ch14      super().__init__(**kwargs)15