BiRefNet-portrait / README.md
ZhengPeng7's picture
Update README.md
db387ea verified
|
raw
history blame
1.64 kB
metadata
license: mit
language:
  - en
pipeline_tag: image-segmentation
tags:
  - dichotomous-image-segmentation
  - salient-object-detection
  - camouflaged-object-detection
  - image-matting

This BiRefNet for portait segmentation is trained on: TR-P3M-10k and validated on TE-P3M-500P and TE-P3M-500NP.

This repo holds the official model weights of "Bilateral Reference for High-Resolution Dichotomous Image Segmentation" (arXiv 2024).

This repo contains the weights of BiRefNet proposed in our paper, which has achieved the SOTA performance on three tasks (DIS, HRSOD, and COD).

Go to my GitHub page for BiRefNet codes and the latest updates: https://github.com/ZhengPeng7/BiRefNet :)

Try our online demos for inference:

  • Inference and evaluation of your given weights: Open In Colab
  • Online Inference with GUI on Hugging Face with adjustable resolutions: Hugging Face Spaces

Citation

@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={arXiv},
  year={2024}
}