ai-models/RMBG-2.0
2
0
Fork 0
mirror of https://www.modelscope.cn/AI-ModelScope/RMBG-2.0.git synced 2026-04-02 11:02:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
ai-modelscope
2024-11-13 12:00:27 +08:00
parent 6b1dc29b74
commit e371aea0f8
12 changed files with 2443 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
BiRefNet_config.py Normal file
View File

@ -0,0 +1,11 @@
from transformers import PretrainedConfig
class BiRefNetConfig(PretrainedConfig):
model_type = "SegformerForSemanticSegmentation"
def __init__(
self,
bb_pretrained=False,
**kwargs
):
self.bb_pretrained = bb_pretrained
super().__init__(**kwargs)