まばらな深度と対応する RGB 画像が与えられた場合、深度補完は、画像全体にまばらな測定値を空間的に伝播して、密な深度予測を取得することを目的としています。深層学習ベースの深さ補完方法の驚異的な進歩にもかかわらず、畳み込み層またはグラフ モデルの局所性により、ネットワークがピクセル間の長期的な関係をモデル化することが困難になります。最近の完全な Transformer ベースのアーキテクチャは、グローバルな受容野で有望な結果を報告していますが、ローカル機能の詳細が悪化しているため、十分に開発された CNN モデルとのパフォーマンスと効率のギャップは依然として存在します。この論文では、畳み込みアテンションレイヤーとビジョントランスフォーマーを1つのブロックに深く結合するJoint Convolutional Attention and Transformerブロック(JCAT)を、ピラミッド構造で深度補完モデルを構築するための基本ユニットとして提案します。このハイブリッド アーキテクチャは、たたみ込みのローカル接続と Transformer のグローバル コンテキストの両方に、1 つのモデルで当然のことながらメリットをもたらします。その結果、当社の CompletionFormer は、屋外の KITTI Depth Completion ベンチマークと屋内の NYUv2 データセットで最先端の CNN ベースの方法よりも優れており、純粋な Transformer ベースの方法と比較して大幅に高い効率 (ほぼ 1/3 FLOP) を達成しています。コードは https://github.com/youmi-zym/CompletionFormer で入手できます。
Given sparse depths and the corresponding RGB images, depth completion aims at spatially propagating the sparse measurements throughout the whole image to get a dense depth prediction. Despite the tremendous progress of deep-learning-based depth completion methods, the locality of the convolutional layer or graph model makes it hard for the network to model the long-range relationship between pixels. While recent fully Transformer-based architecture has reported encouraging results with the global receptive field, the performance and efficiency gaps to the well-developed CNN models still exist because of its deteriorative local feature details. This paper proposes a Joint Convolutional Attention and Transformer block (JCAT), which deeply couples the convolutional attention layer and Vision Transformer into one block, as the basic unit to construct our depth completion model in a pyramidal structure. This hybrid architecture naturally benefits both the local connectivity of convolutions and the global context of the Transformer in one single model. As a result, our CompletionFormer outperforms state-of-the-art CNNs-based methods on the outdoor KITTI Depth Completion benchmark and indoor NYUv2 dataset, achieving significantly higher efficiency (nearly 1/3 FLOPs) compared to pure Transformer-based methods. Code is available at https://github.com/youmi-zym/CompletionFormer.