库名称: birefnet 标签:

• 背景去除
• 掩模生成
• 图像抠图
• pytorch_model_hub_mixin
• model_hub_mixin 仓库地址: https://github.com/ZhengPeng7/BiRefNet 流水线标签: 图像分割

双边参考的高分辨率二分图像分割

                 

本仓库包含BiRefNet用于通用抠图的官方权重。

训练集:

• P3M-10k (除TE-P3M-500-P外)
• TR-humans

验证集:

• TE-P3M-500-P

性能:

查看主BiRefNet模型仓库获取更多信息及使用方法:
https://huggingface.co/ZhengPeng7/BiRefNet/blob/main/README.md

也可访问BiRefNet的GitHub仓库获取所需内容:
https://github.com/ZhengPeng7/BiRefNet

致谢:

• 特别感谢@fal为训练此肖像抠图BiRefNet提供的GPU资源支持。

引用

@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},
  volume = {3},
  pages = {9150038},
  year={2024}
}