库名称: birefnet
标签:

• 背景去除
• 掩模生成
• 二分图像分割
• 伪装目标检测
• 显著目标检测
• pytorch_model_hub_mixin
• model_hub_mixin
  仓库地址: https://github.com/ZhengPeng7/BiRefNet
  流水线标签: 图像分割
  许可证: mit

该BiRefNet模型使用512x512分辨率图像进行训练，以实现更快、更准确的低分辨率推理。

性能表现:

所有测试均在FP16模式下进行。

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

                   

DIS示例1	DIS示例2

本仓库是论文《双边参考的高分辨率二分图像分割》（CAAI AIR 2024）的官方实现。

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

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

致谢:

• 特别感谢@freepik为训练本模型提供的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}  
}