CVPRNov, 2019

使用图神经网络学习特征匹配的 SuperGlue

TL;DR本文介绍了 SuperGlue,它是一个神经网络,通过联合找到对应和拒绝不可匹配的点,匹配两组局部特征。我们通过解决可微分的最优传输问题来估算分配,其成本由图神经网络预测。我们引入了一种基于注意力的灵活的上下文聚合机制,使 SuperGlue 能够联合推理基础的三维场景和特征分配。与传统的手动设计启发式技术相比,我们的技术通过从图像对的端到端训练来学习几何变换的先验知识和三维世界的规律性。SuperGlue 在挑战性的现实室内和室外环境中的姿态估计任务中优于其他学习方法,并实现了最先进的结果。所提出的方法可以在现代 GPU 上实时匹配,并可以轻松集成到现代 SfM 或 SLAM 系统中。代码和训练好的模型在 https URL 公开。