画像誘導深度補完は、疎深度マップと高品質画像から高密度深度マップを生成するタスクです。このタスクでは、色と深さのモダリティをどのように融合するかが、優れたパフォーマンスを実現する上で重要な役割を果たします。この論文は、2つのモダリティを徹底的に活用して融合するために、色が支配的なブランチと深さが支配的なブランチで構成される2つのブランチのバックボーンを提案します。より具体的には、1つのブランチがカラー画像とスパースデプスマップを入力して、デンスデプスマップを予測します。もう1つのブランチは、スパース深度マップと以前に予測された深度マップを入力として受け取り、高密度深度マップも出力します。 2つのブランチから予測された深度マップは互いに補完的であるため、適応的に融合されます。さらに、3D幾何学的手がかりをエンコードするための単純な幾何学的畳み込み層も提案します。幾何学的にエンコードされたバックボーンは、複数の段階で異なるモダリティの融合を実行し、良好な深度完了結果をもたらします。さらに、拡張および加速されたCSPN ++を実装して、融合深度マップを効率的に改良します。提案されたフルモデルは、提出時にKITTI深度完了オンラインリーダーボードで1位にランクされています。また、トップランクのメソッドのほとんどよりもはるかに高速に推測します。この作業のコードはhttps://github.com/JUGGHM/PENet_ICRA2021で入手できます。
Image guided depth completion is the task of generating a dense depth map from a sparse depth map and a high quality image. In this task, how to fuse the color and depth modalities plays an important role in achieving good performance. This paper proposes a two-branch backbone that consists of a color-dominant branch and a depth-dominant branch to exploit and fuse two modalities thoroughly. More specifically, one branch inputs a color image and a sparse depth map to predict a dense depth map. The other branch takes as inputs the sparse depth map and the previously predicted depth map, and outputs a dense depth map as well. The depth maps predicted from two branches are complimentary to each other and therefore they are adaptively fused. In addition, we also propose a simple geometric convolutional layer to encode 3D geometric cues. The geometric encoded backbone conducts the fusion of different modalities at multiple stages, leading to good depth completion results. We further implement a dilated and accelerated CSPN++ to refine the fused depth map efficiently. The proposed full model ranks 1st in the KITTI depth completion online leaderboard at the time of submission. It also infers much faster than most of the top ranked methods. The code of this work is available at https://github.com/JUGGHM/PENet_ICRA2021.