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 transformers and transformers.js tag for discoverability.
Browse files
README.md
CHANGED
|
@@ -8,6 +8,8 @@ tags:
|
|
| 8 |
- Salient Object Detection
|
| 9 |
- pytorch_model_hub_mixin
|
| 10 |
- model_hub_mixin
|
|
|
|
|
|
|
| 11 |
repo_url: https://github.com/ZhengPeng7/BiRefNet
|
| 12 |
pipeline_tag: image-segmentation
|
| 13 |
license: mit
|
|
@@ -206,7 +208,8 @@ Go to my GitHub page for BiRefNet codes and the latest updates: https://github.c
|
|
| 206 |
|
| 207 |
## Acknowledgement:
|
| 208 |
|
| 209 |
-
+ Many thanks to @
|
|
|
|
| 210 |
+ Many thanks to @not-lain for his help on the better deployment of our BiRefNet model on HuggingFace.
|
| 211 |
|
| 212 |
|
|
|
|
| 8 |
- Salient Object Detection
|
| 9 |
- pytorch_model_hub_mixin
|
| 10 |
- model_hub_mixin
|
| 11 |
+
- transformers
|
| 12 |
+
- transformers.js
|
| 13 |
repo_url: https://github.com/ZhengPeng7/BiRefNet
|
| 14 |
pipeline_tag: image-segmentation
|
| 15 |
license: mit
|
|
|
|
| 208 |
|
| 209 |
## Acknowledgement:
|
| 210 |
|
| 211 |
+
+ Many thanks to @Freepik for their generous support on GPU resources for training higher resolution BiRefNet models and more of my explorations.
|
| 212 |
+
+ Many thanks to @fal for their generous support on GPU resources for training better general BiRefNet models.
|
| 213 |
+ Many thanks to @not-lain for his help on the better deployment of our BiRefNet model on HuggingFace.
|
| 214 |
|
| 215 |
|