Image Segmentation
BiRefNet
Safetensors
Transformers
background-removal
mask-generation
Dichotomous Image Segmentation
Camouflaged Object Detection
Salient Object Detection
pytorch_model_hub_mixin
model_hub_mixin
custom_code
Instructions to use ZhengPeng7/BiRefNet with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- BiRefNet
How to use ZhengPeng7/BiRefNet with BiRefNet:
# Option 1: use with transformers from transformers import AutoModelForImageSegmentation birefnet = AutoModelForImageSegmentation.from_pretrained("ZhengPeng7/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("ZhengPeng7/BiRefNet") - Transformers
How to use ZhengPeng7/BiRefNet with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-segmentation", model="ZhengPeng7/BiRefNet", trust_remote_code=True)# pip install -U transformers accelerate # Load model directly from transformers import AutoModelForImageSegmentation model = AutoModelForImageSegmentation.from_pretrained("ZhengPeng7/BiRefNet", trust_remote_code=True, device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit ·
282961d
1
Parent(s): 4e60c70
Add inference endpoint feature in HF model page.
Browse files- birefnet.py +2 -1
birefnet.py
CHANGED
|
@@ -1995,7 +1995,8 @@ class BiRefNet(
|
|
| 1995 |
):
|
| 1996 |
config_class = BiRefNetConfig
|
| 1997 |
def __init__(self, bb_pretrained=True, config=BiRefNetConfig()):
|
| 1998 |
-
super(BiRefNet, self).__init__()
|
|
|
|
| 1999 |
self.config = Config()
|
| 2000 |
self.epoch = 1
|
| 2001 |
self.bb = build_backbone(self.config.bb, pretrained=bb_pretrained)
|
|
|
|
| 1995 |
):
|
| 1996 |
config_class = BiRefNetConfig
|
| 1997 |
def __init__(self, bb_pretrained=True, config=BiRefNetConfig()):
|
| 1998 |
+
super(BiRefNet, self).__init__(config)
|
| 1999 |
+
bb_pretrained = config.bb_pretrained
|
| 2000 |
self.config = Config()
|
| 2001 |
self.epoch = 1
|
| 2002 |
self.bb = build_backbone(self.config.bb, pretrained=bb_pretrained)
|