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

SDformer:用于深度补全的高效端到端Transformer

SDformer: Efficient End-to-End Transformer for Depth Completion

Fudan University(复旦大学);PhotonIC Technologies(光子IC科技公司)

Jian Qian, Miao Sun, Ashley Lee, Jie Li, Shenglong Zhuo, Patrick Yin Chiang

AI总结:

针对CNN深度补全模型感受野受限、标准Transformer计算成本过高的问题,提出基于窗口的SDformer架构,通过多窗口提取长程依赖,在两个基准数据集上以更低开销取得最优性能。

摘要:

深度补全旨在利用深度传感器的稀疏深度测量值预测稠密深度图。目前,基于卷积神经网络(CNN)的模型是深度补全任务中最常用的方法。然而,这类模型尽管高端性能优异,但存在感受野受限的问题。为克服CNN的缺陷,一种更高效强大的方法——Transformer被提出,它是一种采用自适应自注意力机制的序列到序列模型。但标准Transformer的计算成本会随输入分辨率的键-查询点积呈二次增长,并不适用于深度补全任务。\n本研究提出了一种适用于深度补全任务的新型基于窗口的Transformer架构,名为稀疏到稠密Transformer(SDformer)。该网络由用于深度图与RGB图像特征提取及拼接的输入模块、用于提取深层特征的U形编解码器Transformer,以及一个细化模块组成。具体而言,我们首先通过输入模块将深度图特征与RGB图像特征拼接;随后,不对整个特征图计算自注意力,而是采用不同窗口尺寸提取长程深度依赖关系;最后,我们对输入模块和U形编解码器Transformer模块输出的预测特征进行细化,得到丰富的深度特征,再通过卷积层得到稠密深度图。\n实验结果表明,在NYU Depth V2和KITTI DC数据集上,SDformer以更低的计算负载和参数量,取得了优于基于CNN的深度补全模型的当前最优结果。

英文摘要:

Depth completion aims to predict dense depth maps with sparse depth measurements from a depth sensor. Currently, Convolutional Neural Network (CNN) based models are the most popular methods applied to depth completion tasks. However, despite the excellent high-end performance, they suffer from a limited representation area. To overcome the drawbacks of CNNs, a more effective and powerful method has been presented: the Transformer, which is an adaptive self-attention setting sequence-to-sequence model. While the standard Transformer quadratically increases the computational cost from the key-query dot-product of input resolution which improperly employs depth completion tasks. In this work, we propose a different window-based Transformer architecture for depth completion tasks named Sparse-to-Dense Transformer (SDformer). The network consists of an input module for the depth map and RGB image features extraction and concatenation, a U-shaped encoder-decoder Transformer for extracting deep features, and a refinement module. Specifically, we first concatenate the depth map features with the RGB image features through the input model. Then, instead of calculating self-attention with the whole feature maps, we apply different window sizes to extract the long-range depth dependencies. Finally, we refine the predicted features from the input module and the U-shaped encoder-decoder Transformer module to get the enriching depth features and employ a convolution layer to obtain the dense depth map. In practice, the SDformer obtains state-of-the-art results against the CNN-based depth completion models with lower computing loads and parameters on the NYU Depth V2 and KITTI DC datasets.

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

↑