Gated Recurrent Fusion UNet for Depth Completion

Tao Li,Xiucheng Dong,Hongwei Lin
DOI: https://doi.org/10.1007/s11063-023-11334-w
IF: 2.565
2023-01-01
Neural Processing Letters
Abstract:Recently, image guided depth completion has been widely explored and obtained remarkable performance. However, most of the existing image guided depth completion methods often fuse multilevel features or multimodal features by performing simple element-wise addition or concatenation, hence hindering the fusion effectiveness. To address this issue, we propose a gated recurrent fusion UNet for more effective feature fusion. Specifically, a gated recurrent model is used to adaptively select and fuse useful information from multilevel features. Moreover, we introduce a dimension-wise attention based CSPN++ module to iteratively refine depth estimation, which incorporates dimension-wise attention to fully learn feature interdependencies to obtain a more accurate neighborhood affinity matrix for spatial propagation. Comprehensive experimental results demonstrate that the proposed method outperforms many state-of-the-art methods in terms of both quantitative evaluations and visual qualities.
What problem does this paper attempt to address?