Efficient Depth Completion Network Based on Dynamic Gated Fusion

Zhengyang Mu,Qi,Jingyu Wang,Haifeng Sun,Jianxin Liao
DOI: https://doi.org/10.1007/978-3-030-82153-1_24
2021-01-01
Abstract:Depth completion aims to recover dense depth maps from sparse depth and RGB images. Current methods achieve high accuracy at the cost of large model size and huge computation complexity, which prevent them from wider applications. In this paper, we focus on making two key issues on depth completion - feature extraction and fusion more efficient to achieve superior trade-off in model size and accuracy: (1) we propose efficient dual-branch encoder by exploring data characteristics of different modalities which can greatly reduce the model size and inference time; (2) we propose a dynamic gated fusion module, which is guided by input sparse depth to fuse information of both RGB and sparse depth feature more efficiently by generating dynamic fusing weights. Experiments on KITTI Depth Completion and NYU Depth v2 show that our method achieves 3.5x - 10x speedup against the state-of-art method, 9x param compressing and comparable accuracy compared with state-of-the-art methods, which shows our method achieves good trade-off between performance and speed.
What problem does this paper attempt to address?