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)# Load model directly from transformers import AutoModelForImageSegmentation model = AutoModelForImageSegmentation.from_pretrained("ZhengPeng7/BiRefNet", trust_remote_code=True, dtype="auto") - Notebooks
- Google Colab
- Kaggle
Commit ·
fc65c0f
1
Parent(s): 0e81815
For users to load in one key.
Browse files- hf_config.py → BiRefNet_config.py +0 -0
- config.json +1 -0
hf_config.py → BiRefNet_config.py
RENAMED
|
File without changes
|
config.json
CHANGED
|
@@ -4,6 +4,7 @@
|
|
| 4 |
"BiRefNet"
|
| 5 |
],
|
| 6 |
"auto_map": {
|
|
|
|
| 7 |
"AutoModelForImageSegmentation": "BiRefNet_github.models.birefnet.BiRefNet"
|
| 8 |
},
|
| 9 |
"custom_pipelines": {
|
|
|
|
| 4 |
"BiRefNet"
|
| 5 |
],
|
| 6 |
"auto_map": {
|
| 7 |
+
"AutoConfig": "BiRefNet_config.BiRefNetConfig",
|
| 8 |
"AutoModelForImageSegmentation": "BiRefNet_github.models.birefnet.BiRefNet"
|
| 9 |
},
|
| 10 |
"custom_pipelines": {
|