An Adaptive Framework for Learning Unsupervised Depth Completion

被引:28
作者
Wong, Alex [1 ]
Fei, Xiaohan [1 ,2 ]
Hong, Byung-Woo [3 ]
Soatto, Stefano [1 ]
机构
[1] Univ Calif Los Angeles, Dept Comp Sci, Los Angeles, CA 90095 USA
[2] Amazon Web Serv, Seattle, WA 98109 USA
[3] Chung Ang Univ, Dept Comp Sci, Seoul 06973, South Korea
关键词
Training; Optimization; Data models; Adaptation models; Uncertainty; Image reconstruction; Computer science; Sensor fusion; visual learning; IMAGE-RESTORATION; REGULARIZATION;
D O I
10.1109/LRA.2021.3062602
中图分类号
TP24 [机器人技术];
学科分类号
080202 ; 1405 ;
摘要
We present a method to infer a dense depth map from a color image and associated sparse depth measurements. Our main contribution lies in the design of an annealing process for determining co-visibility (occlusions, disocclusions) and the degree of regularization to impose on the model. We show that regularization and co-visibility are related via the fitness (residual) of model to data and both can be unified into a single framework to improve the learning process. Our method is an adaptive weighting scheme that guides optimization by measuring the residual at each pixel location over each training step for (i) estimating a soft visibility mask and (ii) determining the amount of regularization. We demonstrate the effectiveness our method by applying it to several recent unsupervised depth completion methods and improving their performance on public benchmark datasets, without incurring additional trainable parameters or increase in inference time.
引用
收藏
页码:3120 / 3127
页数:8
相关论文
共 32 条
[1]  
Chen Y, 2019, IEEE I CONF COMP VIS, P10022, DOI [10.1109/ICCV.2019.01012, 10.1109/iccv.2019.01012]
[2]  
Chodosh N., 2018, ASIAN C COMPUTER VIS, P499
[3]   Learning Morphological Operators for Depth Completion [J].
Dimitrievski, Martin ;
Veelaert, Peter ;
Philips, Wilfried .
ADVANCED CONCEPTS FOR INTELLIGENT VISION SYSTEMS, ACIVS 2018, 2018, 11182 :450-461
[4]   DeepPruner: Learning Efficient Stereo Matching via Differentiable PatchMatch [J].
Duggal, Shivam ;
Wang, Shenlong ;
Ma, Wei-Chiu ;
Hu, Rui ;
Urtasun, Raquel .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :4383-4392
[5]   Geo-Supervised Visual Depth Prediction [J].
Fei, Xiaohan ;
Wong, Alex ;
Soatto, Stefano .
IEEE ROBOTICS AND AUTOMATION LETTERS, 2019, 4 (02) :1661-1668
[6]   Methods for choosing the regularization parameter and estimating the noise variance in image restoration and their relation [J].
Galatsanos, Nikolas P. ;
Katsaggelos, Aggelos K. .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 1992, 1 (03) :322-336
[7]   Adaptive Regularization of Some Inverse Problems in Image Analysis [J].
Hong, Byung-Woo ;
Koo, Ja-Keoung ;
Burger, Martin ;
Soatto, Stefano .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2020, 29 :2507-2521
[8]   Adaptive Regularization in Convex Composite Optimization for Variational Imaging Problems [J].
Hong, Byung-Woo ;
Koo, Ja-Keoung ;
Dirks, Hendrik ;
Burger, Martin .
PATTERN RECOGNITION (GCPR 2017), 2017, 10496 :268-280
[9]   A Quantitative Evaluation of Confidence Measures for Stereo Vision [J].
Hu, Xiaoyan ;
Mordohai, Philippos .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2012, 34 (11) :2121-2133
[10]   HMS-Net: Hierarchical Multi-Scale Sparsity-Invariant Network for Sparse Depth Completion [J].
Huang, Zixuan ;
Fan, Junming ;
Cheng, Shenggan ;
Yi, Shuai ;
Wang, Xiaogang ;
Li, Hongsheng .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2020, 29 :3429-3441