briaai
/

RMBG-1.4

Image Segmentation

SegformerForSemanticSegmentation

remove background

background-removal

legal liability

Model card Files Files and versions Community

OriLib commited on Mar 26

Commit

e78c703

•

1 Parent(s): 83045c9

Update briarmbg.py

Files changed (1) hide show

briarmbg.py +1 -1

briarmbg.py CHANGED Viewed

@@ -348,7 +348,7 @@ class myrebnconv(nn.Module):
 class BriaRMBG(PreTrainedModel):
     config_class = RMBGConfig
-    def __init__(self,config):
         super().__init__(config)
         in_ch = config.in_ch # 3
         out_ch = config.out_ch # 1

 class BriaRMBG(PreTrainedModel):
     config_class = RMBGConfig
+    def __init__(self,config:RMBGConfig = RMBGConfig()):
         super().__init__(config)
         in_ch = config.in_ch # 3
         out_ch = config.out_ch # 1