通过分层注意力机制提升特征点匹配精度,有效抑制大量异常值干扰。
LLHA-Net: A Hierarchical Attention Network for Two-View Correspondence Learning
- 采用分层通道融合与注意力模块,增强特征语义表达能力。
- 在YFCC100M和SUN3D数据集上显著优于现有方法,提升匹配准确率。
- 适合需要高鲁棒性特征匹配的视觉定位与三维重建任务。
建立特征点间的正确对应关系是计算机视觉的基础任务。然而,特征点中存在大量异常值会严重影响匹配结果,降低过程的准确性与鲁棒性。尤其当异常值比例较高时,如何在保留高质量信息的同时减少负样本带来的误差成为关键挑战。为此,本文提出一种名为逐层分层注意力网络(LLHA-Net)的新方法,通过阶段融合、分层提取与注意力机制提升特征点匹配精度。我们设计了逐层通道融合模块,保留各阶段的语义信息并实现整体融合,增强特征表达能力;同时引入分层注意力模块,自适应融合全局感知与结构语义信息。此外,提出了两种特征提取与集成架构,提升网络适应性。在YFCC100M和SUN3D两个公开数据集上的实验表明,该方法在异常值剔除与相机位姿估计方面均优于多个当前最优技术。源代码已公开于http://www.linshuyuan.com。
原文摘要 · Abstract (English)
Establishing the correct correspondence of feature points is a fundamental task in computer vision. However, the presence of numerous outliers among the feature points can significantly affect the matching results, reducing the accuracy and robustness of the process. Furthermore, a challenge arises when dealing with a large proportion of outliers: how to ensure the extraction of high-quality information while reducing errors caused by negative samples. To address these issues, in this paper, we propose a novel method called Layer-by-Layer Hierarchical Attention Network, which enhances the precision of feature point matching in computer vision by addressing the issue of outliers. Our method incorporates stage fusion, hierarchical extraction, and an attention mechanism to improve the network's representation capability by emphasizing the rich semantic information of feature points. Specifically, we introduce a layer-by-layer channel fusion module, which preserves the feature semantic information from each stage and achieves overall fusion, thereby enhancing the representation capability of the feature points. Additionally, we design a hierarchical attention module that adaptively captures and fuses global perception and structural semantic information using an attention mechanism. Finally, we propose two architectures to extract and integrate features, thereby improving the adaptability of our network. We conduct experiments on two public datasets, namely YFCC100M and SUN3D, and the results demonstrate that our proposed method outperforms several state-of-the-art techniques in both outlier removal and camera pose estimation. Source code is available at http://www.linshuyuan.com.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。