Learning to Match Features with Seeded Graph Matching Network

被引:67
作者
Chen, Hongkai [1 ]
Luo, Zixin [1 ]
Zhang, Jiahui [2 ]
Zhou, Lei [1 ]
Bai, Xuyang [1 ]
Hu, Zeyu [1 ]
Tai, Chiew-Lan [1 ]
Quan, Long [1 ]
机构
[1] Hong Kong Univ Sci & Technol, Hong Kong, Peoples R China
[2] Tsinghua Univ, Beijing, Peoples R China
来源
2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021) | 2021年
关键词
D O I
10.1109/ICCV48922.2021.00624
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Matching local features across images is a fundamental problem in computer vision. Targeting towards high accuracy and efficiency, we propose Seeded Graph Matching Network, a graph neural network with sparse structure to reduce redundant connectivity and learn compact representation. The network consists of 1) Seeding Module, which initializes the matching by generating a small set of reliable matches as seeds. 2) Seeded Graph Neural Network, which utilizes seed matches to pass messages within/across images and predicts assignment costs. Three novel operations are proposed as basic elements for message passing: 1) Attentional Pooling, which aggregates keypoint features within the image to seed matches. 2) Seed Filtering, which enhances seed features and exchanges messages across images. 3) Attentional Unpooling, which propagates seed features back to original keypoints. Experiments show that our method reduces computational and memory complexity significantly compared with typical attention-based networks while competitive or higher performance is achieved.
引用
收藏
页码:6281 / 6290
页数:10
相关论文
共 66 条
[1]  
[Anonymous], 2018, ECCV, DOI DOI 10.1007/978-3-030-01237-3_47
[2]  
[Anonymous], 2018, CVPR, DOI DOI 10.1109/CVPR.2018.00282
[3]  
[Anonymous], 2015, CVPR
[4]  
[Anonymous], 2018, CVPR, DOI DOI 10.1109/CVPR.2018.00218
[5]  
[Anonymous], 2018, NEURIPS
[6]  
[Anonymous], 2020, CVPR, DOI DOI 10.1109/CVPR42600.2020.00662
[7]  
[Anonymous], 2018, 6 INT C LEARN REPR I
[8]  
[Anonymous], 2020, CVPR, DOI DOI 10.1109/CVPR42600.2020.00468
[9]  
[Anonymous], 2014, ECCV
[10]  
[Anonymous], 2018, CVPR, DOI DOI 10.1109/CVPR.2018.00598