D2VT: Better Detection and Description of Local Features with Vision Transformers
Constrained by the local nature of CNNs, existing local feature description methods often overlook global and contextual spatial information. Vision Transformers (ViT) address this by leveraging self-attention to capture long-range dependencies and preserve spatial details more effectively than CNNs. Our work introduces a hybrid architecture that merges CNNs for local feature extraction with ViT for global feature capture, enhancing performance across diverse vision tasks. We propose a novel hierarchical Transformer encoder adaptable to various image resolutions, yielding multi-scale features without positional encoding. Additionally, we introduce a consistent attention-weighted triple loss to get the attention map and to optimize and match local descriptors. Utilizing a feature pyramid, our method predicts keypoints at multiple scales, leading to improved localization accuracy. Experiments have shown that our approach is competitive with the leading contrastive learning methods in image matching benchmarks and demonstrates robust generalization in tasks like visual odometry.
Authors
- Zihao Wang (ORCID: https://orcid.org/0000-0001-6953-0470)
- Zhen Li (ORCID: https://orcid.org/0000-0002-2049-2108)
- Yifei Yang
- Yidian Huang (ORCID: https://orcid.org/0009-0006-8267-6590)
- Fang Deng (ORCID: https://orcid.org/0000-0002-1111-7285)
Institutions
- Beijing Institute of Technology (CN)
- Peking University (CN)
Publication Details
- Published
- 2024-11-01
- DOI
- https://doi.org/10.1109/cac63892.2024.10864608
- Primary Topic
- Industrial Vision Systems and Defect Detection
- Type
- article
- Field-Weighted Citation Impact
- 0.00