Image Segmentation
BiRefNet
Safetensors
background-removal
mask-generation
Dichotomous Image Segmentation
pytorch_model_hub_mixin
model_hub_mixin
custom_code
Instructions to use ZhengPeng7/BiRefNet-DIS5K with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- BiRefNet
How to use ZhengPeng7/BiRefNet-DIS5K with BiRefNet:
# Option 1: use with transformers from transformers import AutoModelForImageSegmentation birefnet = AutoModelForImageSegmentation.from_pretrained("ZhengPeng7/BiRefNet-DIS5K", 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-DIS5K") - Notebooks
- Google Colab
- Kaggle
Update the bibtex.
Browse files
README.md
CHANGED
|
@@ -48,10 +48,12 @@ https://github.com/ZhengPeng7/BiRefNet
|
|
| 48 |
## Citation
|
| 49 |
|
| 50 |
```
|
| 51 |
-
@article{
|
| 52 |
title={Bilateral Reference for High-Resolution Dichotomous Image Segmentation},
|
| 53 |
author={Zheng, Peng and Gao, Dehong and Fan, Deng-Ping and Liu, Li and Laaksonen, Jorma and Ouyang, Wanli and Sebe, Nicu},
|
| 54 |
journal={CAAI Artificial Intelligence Research},
|
|
|
|
|
|
|
| 55 |
year={2024}
|
| 56 |
}
|
| 57 |
```
|
|
|
|
| 48 |
## Citation
|
| 49 |
|
| 50 |
```
|
| 51 |
+
@article{zheng2024birefnet,
|
| 52 |
title={Bilateral Reference for High-Resolution Dichotomous Image Segmentation},
|
| 53 |
author={Zheng, Peng and Gao, Dehong and Fan, Deng-Ping and Liu, Li and Laaksonen, Jorma and Ouyang, Wanli and Sebe, Nicu},
|
| 54 |
journal={CAAI Artificial Intelligence Research},
|
| 55 |
+
volume = {3},
|
| 56 |
+
pages = {9150038},
|
| 57 |
year={2024}
|
| 58 |
}
|
| 59 |
```
|