README.md

---
library_name: nobg
tags:
- birefnet
- model_hub_mixin
- nobg
- pytorch_model_hub_mixin
---

<p align="center">
<img src="https://usefeyn.com/feyn/feyn_mark.svg"/>
</p>

This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration.

Library: [nobg](https://github.com/feyninc/nobg)

## how to load
```
pip install nobg
```

use the AutoModel class
```python
from nobg import AutoModel
model = AutoModel.from_pretrained("nobg/BiRefNet")
```
or you can use the model class directly
```python
from nobg import BiRefNet
model = BiRefNet.from_pretrained("nobg/BiRefNet")
```


## Citation
If you use this model, please cite the original paper
(https://arxiv.org/abs/2401.03407):
```bibtex
@article{zheng2024birefnet,
  title={Bilateral Reference for High-Resolution Dichotomous Image Segmentation},
  author={Zheng, Peng and Gao, Dehong and Fan, Deng-Ping and Liu, Li and
          Laaksonen, Jorma and Ouyang, Wanli and Sebe, Nicu},
  journal={CAAI Artificial Intelligence Research},
  year={2024}
}
```

## Contributions
Any contributions are welcome at https://github.com/feyninc/nobg