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

我们提出了一种新的双边参考框架（BiRefNet），用于高分辨率二值图像分割（DIS）。该框架包含两个核心组件：定位模块（LM）和重建模块（RM），其中我们提出了双边参考（BiRef）。定位模块利用全局语义信息辅助目标定位。在重建模块中，我们利用双边参考进行重建过程，其中图像的层次化块提供源参考，而梯度图则作为目标参考。这些组件协同工作，生成最终的预测图。此外，我们引入了辅助梯度监督机制，以增强对细节更精细区域的关注。为进一步提高地图质量和训练过程，我们还详细介绍了针对二值图像分割的实际训练策略。为了验证我们的方法具有广泛的适用性，我们在四个任务上进行了大量实验，结果表明BiRefNet表现出显著的性能，在所有基准测试中均优于特定任务的最先进方法。我们的代码已发布在 https://github.com/ZhengPeng7/BiRefNet。