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
Add license (#4)
Browse files- Add license (ef49e65518c1273f50e879a504c622e3be01bd27)
Co-authored-by: Lain <not-lain@users.noreply.huggingface.co>
README.md
CHANGED
|
@@ -10,6 +10,7 @@ tags:
|
|
| 10 |
- model_hub_mixin
|
| 11 |
repo_url: https://github.com/ZhengPeng7/BiRefNet
|
| 12 |
pipeline_tag: image-segmentation
|
|
|
|
| 13 |
---
|
| 14 |
<h1 align="center">Bilateral Reference for High-Resolution Dichotomous Image Segmentation</h1>
|
| 15 |
|
|
@@ -171,4 +172,4 @@ Go to my GitHub page for BiRefNet codes and the latest updates: https://github.c
|
|
| 171 |
journal={CAAI Artificial Intelligence Research},
|
| 172 |
year={2024}
|
| 173 |
}
|
| 174 |
-
```
|
|
|
|
| 10 |
- model_hub_mixin
|
| 11 |
repo_url: https://github.com/ZhengPeng7/BiRefNet
|
| 12 |
pipeline_tag: image-segmentation
|
| 13 |
+
license: mit
|
| 14 |
---
|
| 15 |
<h1 align="center">Bilateral Reference for High-Resolution Dichotomous Image Segmentation</h1>
|
| 16 |
|
|
|
|
| 172 |
journal={CAAI Artificial Intelligence Research},
|
| 173 |
year={2024}
|
| 174 |
}
|
| 175 |
+
```
|