Bilateral Reference for High-Resolution Dichotomous Image Segmentation
Paper • 2401.03407 • Published • 4
# Option 2: use with BiRefNet
# Install from https://github.com/ZhengPeng7/BiRefNet
from models.birefnet import BiRefNet
model = BiRefNet.from_pretrained("nobg/BiRefNet")
This model has been pushed to the Hub using the PytorchModelHubMixin integration.
Library: nobg
how to load
pip install nobg
use the AutoModel class
from nobg import AutoModel
model = AutoModel.from_pretrained("nobg/BiRefNet")
or you can use the model class directly
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):
@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
- Downloads last month
- 46
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support
# Option 1: use with transformers from transformers import AutoModelForImageSegmentation birefnet = AutoModelForImageSegmentation.from_pretrained("nobg/BiRefNet", trust_remote_code=True)