用于高分辨率二值图像分割的双向参考框架
计算机视觉与模式识别
2025-07-28 v7
摘要
我们提出了一种用于高分辨率二值图像分割（DIS）的新型双向参考框架（BiRefNet）。它包含两个核心组件：定位模块（LM）和我们提出的具有双向参考（BiRef）的重建模块（RM）。LM 利用全局语义信息辅助物体定位。在 RM 中，我们利用 BiRef 进行重建过程，其中图像的层次化补丁提供源参考，梯度图作为目标参考。这些组件协同工作以生成最终的预测图。我们还引入了辅助梯度监督，以增强对具有更精细细节区域的关注。此外，我们概述了专为 DIS 设计的实用训练策略，以提高地图质量和训练过程。为了验证我们方法的通用适用性，我们在四个任务上进行了广泛的实验，表明 BiRefNet 表现出卓越的性能，在所有基准测试中均优于特定任务的最先进方法。我们的代码可在 https://github.com/ZhengPeng7/BiRefNet 获取。
引用
@article{arxiv.2401.03407,
title = {Bilateral Reference for High-Resolution Dichotomous Image Segmentation},
author = {Peng Zheng and Dehong Gao and Deng-Ping Fan and Li Liu and Jorma Laaksonen and Wanli Ouyang and Nicu Sebe},
journal= {arXiv preprint arXiv:2401.03407},
year = {2025}
}
备注
Version 7, fix the A/B reverse problem in Fig. 9