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

基于结构注意记忆网络的单目深度估计

Structure-Attentioned Memory Network for Monocular Depth Estimation

New York University(纽约大学);New York University Abu Dhabi(纽约大学阿布扎比分校)

Jing Zhu, Yunxiao Shi, Mengwei Ren, Yi Fang, Kuo-Chin Lien, Junli Gu

摘要:

单目深度估计是一项具有挑战性的任务,其目的是从给定的单个RGB图像预测对应的深度图。最近的深度学习模型已经被提出来通过学习RGB图像和深度域之间的深度特征的对准来预测来自图像的深度。在本文中,我们提出了一种新的方法,称为结构注意记忆网络,通过考虑到常见的结构特征(例如,重复结构模式、平面、对称性)。为此,我们引入了一个新的面向结构的记忆(SOM)模块来学习和记忆RGB图像域和深度域之间的结构特定信息。更具体地说,在SOM模块中,我们开发了一个可记忆的银行过滤器(MBF)单元学习一组过滤器,记住结构感知的图像深度残差模式,以及注意力引导控制器(AGC)单元,以控制过滤器的选择在MBF给定的图像特征查询。给定查询图像特征,训练好的SOM模块能够自适应地选择最佳的自定义滤波器用于跨域特征传输,具有图像和深度之间的最佳结构差异。总之,我们专注于解决这个结构特定的域适应的挑战,提出了一种新的端到端的多尺度难忘的网络,单目深度估计。实验表明,我们提出的模型表现出优越的性能相比,现有的监督单目深度估计方法上具有挑战性的KITTI和纽约大学深度V2基准。

英文摘要:

Monocular depth estimation is a challenging task that aims to predict a corresponding depth map from a given single RGB image. Recent deep learning models have been proposed to predict the depth from the image by learning the alignment of deep features between the RGB image and the depth domains. In this paper, we present a novel approach, named Structure-Attentioned Memory Network, to more effectively transfer domain features for monocular depth estimation by taking into account the common structure regularities (e.g., repetitive structure patterns, planar surfaces, symmetries) in domain adaptation. To this end, we introduce a new Structure-Oriented Memory (SOM) module to learn and memorize the structure-specific information between RGB image domain and the depth domain. More specifically, in the SOM module, we develop a Memorable Bank of Filters (MBF) unit to learn a set of filters that memorize the structure-aware image-depth residual pattern, and also an Attention Guided Controller (AGC) unit to control the filter selection in the MBF given image features queries. Given the query image feature, the trained SOM module is able to adaptively select the best customized filters for cross-domain feature transferring with an optimal structural disparity between image and depth. In summary, we focus on addressing this structure-specific domain adaption challenge by proposing a novel end-to-end multi-scale memorable network for monocular depth estimation. The experiments show that our proposed model demonstrates the superior performance compared to the existing supervised monocular depth estimation approaches on the challenging KITTI and NYU Depth V2 benchmarks.

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

↑