库名称: birefnet 标签:

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

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

                 

本仓库包含BiRefNet通用抠图模型的官方权重

训练集:

• P3M-10k (除TE-P3M-500-NP外)
• TR-humans
• AM-2k
• AIM-500
• Human-2k (与BG-20k合成)
• Distinctions-646 (与BG-20k合成)
• HIM2K
• PPM-100

验证集:

• TE-P3M-500-NP

性能表现:

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

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

致谢:

• 特别感谢@freepik为本模型训练提供的GPU资源支持！

引用文献

@article{zheng2024birefnet,
  title={双边参考的高分辨率二分图像分割},
  author={郑鹏 and 高德宏 and 范登平 and 刘力 and Jorma Laaksonen and 欧阳万里 and Nicu Sebe},
  journal={CAAI人工智能研究},
  volume = {3},
  pages = {9150038},
  year={2024}
}