arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2211.05783cs.CV

统一光流、立体匹配与深度估计

Unifying Flow, Stereo and Depth Estimation

ETH Zurich(苏黎世联邦理工学院);University of Tübingen(图宾根大学);Sydney AI Center(悉尼人工智能中心);The University of Sydney(悉尼大学);Monash University(莫纳什大学);Max Planck Institute for Intelligent Systems(马克斯·普朗克智能系统研究所)

Haofei Xu, Jing Zhang, Jianfei Cai, Hamid Rezatofighi, Fisher Yu, Dacheng Tao, Andreas Geiger

AI总结:

本文提出统一公式与单一Transformer模型,将光流、立体匹配和深度估计统一为密集对应匹配问题,利用交叉注意力提升特征质量,实现跨任务迁移,并在多个基准上超越或媲美现有方法。

摘要:

我们提出了一种统一的公式和模型,用于三项运动和3D感知任务:光流估计、校正立体匹配以及基于姿态图像的非校正立体深度估计。与以往针对每个特定任务设计的专用架构不同,我们将这三项任务统一表述为一个密集对应匹配问题,可以通过直接比较特征相似度,使用单一模型来解决。这种公式要求具有判别性的特征表示,我们通过Transformer,特别是交叉注意力机制来实现。我们证明,交叉注意力能够通过跨视图交互整合来自另一幅图像的知识,从而极大地提升提取特征的质量。由于模型架构和参数在各任务间共享,我们的统一模型自然支持跨任务迁移。在具有挑战性的Sintel数据集上,我们的统一模型优于RAFT;我们的最终模型仅使用少量额外的任务特定细化步骤,在10个流行的光流、立体和深度数据集上,性能优于或可与近期最先进的方法相媲美,同时在模型设计和推理速度方面更加简单和高效。

英文摘要:

We present a unified formulation and model for three motion and 3D perception tasks: optical flow, rectified stereo matching and unrectified stereo depth estimation from posed images. Unlike previous specialized architectures for each specific task, we formulate all three tasks as a unified dense correspondence matching problem, which can be solved with a single model by directly comparing feature similarities. Such a formulation calls for discriminative feature representations, which we achieve using a Transformer, in particular the cross-attention mechanism. We demonstrate that cross-attention enables integration of knowledge from another image via cross-view interactions, which greatly improves the quality of the extracted features. Our unified model naturally enables cross-task transfer since the model architecture and parameters are shared across tasks. We outperform RAFT with our unified model on the challenging Sintel dataset, and our final model that uses a few additional task-specific refinement steps outperforms or compares favorably to recent state-of-the-art methods on 10 popular flow, stereo and depth datasets, while being simpler and more efficient in terms of model design and inference speed.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑