arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2504.04701cs.CV

DFormerv2:用于RGBD语义分割的几何自注意力

DFormerv2: Geometry Self-Attention for RGBD Semantic Segmentation

Nankai University(南开大学)

Bo-Wen Yin, Jiao-Long Cao, Ming-Ming Cheng, Qibin Hou

AI总结:

本文提出DFormerv2,将深度图作为几何先验而非用神经网络编码,通过从深度和图像块空间距离中提取几何线索来指导自注意力权重分配,在多个RGBD语义分割基准上取得卓越性能。

摘要:

近年来,场景理解的进展在很大程度上受益于深度图所提供的三维几何信息,尤其是在低光照和过曝光等复杂条件下。现有方法将深度图与RGB图像一起编码,并在二者之间进行特征融合,以获得更鲁棒的预测。考虑到深度可以被视为RGB图像的几何补充,一个直接的问题随之而来:我们是否真的需要像处理RGB图像那样,用神经网络显式地编码深度信息?基于这一见解,本文研究了一种学习RGBD特征表示的新方法,并提出了DFormerv2,这是一种强大的RGBD编码器,它显式地将深度图用作几何先验,而不是用神经网络编码深度信息。我们的目标是从深度以及所有图像块token之间的空间距离中提取几何线索,然后将其用作几何先验,在自注意力中分配注意力权重。大量实验表明,DFormerv2在各种RGBD语义分割基准上表现出卓越的性能。代码可在以下地址获取:https://github.com/VCIP-RGBD/DFormer。

英文摘要:

Recent advances in scene understanding benefit a lot from depth maps because of the 3D geometry information, especially in complex conditions (e.g., low light and overexposed). Existing approaches encode depth maps along with RGB images and perform feature fusion between them to enable more robust predictions. Taking into account that depth can be regarded as a geometry supplement for RGB images, a straightforward question arises: Do we really need to explicitly encode depth information with neural networks as done for RGB images? Based on this insight, in this paper, we investigate a new way to learn RGBD feature representations and present DFormerv2, a strong RGBD encoder that explicitly uses depth maps as geometry priors rather than encoding depth information with neural networks. Our goal is to extract the geometry clues from the depth and spatial distances among all the image patch tokens, which will then be used as geometry priors to allocate attention weights in self-attention. Extensive experiments demonstrate that DFormerv2 exhibits exceptional performance in various RGBD semantic segmentation benchmarks. Code is available at: https://github.com/VCIP-RGBD/DFormer.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑