Duplicated from ZhengPeng7/BiRefNet
How to use not-lain/BiRefNet with BiRefNet:
# Option 1: use with transformers from transformers import AutoModelForImageSegmentation birefnet = AutoModelForImageSegmentation.from_pretrained("not-lain/BiRefNet", trust_remote_code=True)
# Option 2: use with BiRefNet # Install from https://github.com/ZhengPeng7/BiRefNet from models.birefnet import BiRefNet model = BiRefNet.from_pretrained("not-lain/BiRefNet")
74fddf5
1
2
3
4
5
6
7
8
9
10
11
12
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)
