Multi-condition guided depth map completion method based on diffusion model
Ren Qian, Wenfeng Qiu, Wenbang Yang, Jianhua Li, Xin’an Wang, Yong Jun Zhao · 2023
The diffusion model has been widely applied in various aspects of artificial intelligence due to its flexible and diverse generative performance. However, there is a lack of research on applying diffusion models in the field of depth map restoration. This is primarily because the diverse generation capabilities are not sufficient to meet the requirements for depth map completion. Depth map completion requires rational completion in the current scene. In order to adapt diffusion models for depth map completion tasks, this paper proposes the Multi Condition Diffusion Model (MCDM). It allows the addition of conditional information to constrain the model’s rational completion of ill-regions. The MultiConditionLN module effectively adds multiple conditions to the depth map completion task. This module uses the completion region mask and the input image as conditions to constrain the model’s generation process. This enables the model to complete the regions that need restoration based on the scene of the input image. The proposed model achieves promising results on depth map datasets.