arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2204.13635cs.CV

SemAttNet:基于注意力的语义感知引导深度补全

SemAttNet: Towards Attention-based Semantic Aware Guided Depth Completion

Department of Computer Science, University of Kaiserslautern, 67663 Kaiserslautern, Germany(凯斯莱特大学计算机科学系);Mindgrage, University of Kaiserslautern, 67663 Kaiserslautern, Germany(Mindgrage凯斯莱特大学);Department of Computer Science, Luleå University of Technology, 971 87 Luleå, Sweden(卢勒奥技术大学计算机科学系)

Danish Nazir, Marcus Liwicki, Didier Stricker, Muhammad Zeshan Afzal

AI总结:

本文提出SemAttNet,通过三分支架构结合颜色、语义和深度引导,利用注意力机制融合特征,提升深度补全精度,实验表明在KITTI基准上达到最优性能。

摘要:

深度补全涉及从稀疏图和RGB图像中恢复密集深度图。最近的方法侧重于利用颜色图像作为指导图像来恢复无效像素的深度。然而,单独的颜色图像不足以提供场景的必要语义理解。因此,深度补全任务受到RGB图像中突然光照变化(如阴影)的限制。本文提出了一种新的三分支主干,包括颜色引导、语义引导和深度引导分支。具体而言,颜色引导分支以稀疏深度图和RGB图像为输入,生成包含场景颜色线索(如物体边界)的颜色深度。颜色引导分支预测的密集深度图与语义图像和稀疏深度图一起作为输入传递给语义引导分支以估计语义深度。深度引导分支利用稀疏、颜色和语义深度生成密集深度图。颜色深度、语义深度和引导深度被自适应融合以产生我们提出的三分支主干的输出。此外,我们还提出应用语义感知多模态注意力融合块(SAMMAFB)来融合所有三个分支之间的特征。我们进一步使用CSPN++与空洞卷积来细化由我们三分支主干生成的密集深度图。广泛的实验表明,我们的模型在提交时的KITTI深度补全基准上实现了最先进的性能。

英文摘要:

Depth completion involves recovering a dense depth map from a sparse map and an RGB image. Recent approaches focus on utilizing color images as guidance images to recover depth at invalid pixels. However, color images alone are not enough to provide the necessary semantic understanding of the scene. Consequently, the depth completion task suffers from sudden illumination changes in RGB images (e.g., shadows). In this paper, we propose a novel three-branch backbone comprising color-guided, semantic-guided, and depth-guided branches. Specifically, the color-guided branch takes a sparse depth map and RGB image as an input and generates color depth which includes color cues (e.g., object boundaries) of the scene. The predicted dense depth map of color-guided branch along-with semantic image and sparse depth map is passed as input to semantic-guided branch for estimating semantic depth. The depth-guided branch takes sparse, color, and semantic depths to generate the dense depth map. The color depth, semantic depth, and guided depth are adaptively fused to produce the output of our proposed three-branch backbone. In addition, we also propose to apply semantic-aware multi-modal attention-based fusion block (SAMMAFB) to fuse features between all three branches. We further use CSPN++ with Atrous convolutions to refine the dense depth map produced by our three-branch backbone. Extensive experiments show that our model achieves state-of-the-art performance in the KITTI depth completion benchmark at the time of submission.

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

↑