Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Learning to Match Features with Seeded Graph Matching Network

About

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.

Hongkai Chen, Zixin Luo, Jiahui Zhang, Lei Zhou, Xuyang Bai, Zeyu Hu, Chiew-Lan Tai, Long Quan• 2021

Related benchmarks

TaskDatasetResultRank
Relative Pose EstimationMegaDepth (test)
Pose AUC @5°40.5
83
Homography EstimationHPatches
Overall Accuracy (< 1px)52
59
Pose EstimationYFCC100M
AUC (5°)35.63
28
Pose EstimationMegaDepth 1500 (test)--
27
Visual LocalizationAachen Day-Night v1.1 (test)
Success Rate (0.25m, 2°)72.3
24
Relative Pose EstimationYFCC100m v1.0 (test)
AUC @ 5°34.8
22
Relative Pose EstimationScanNet
AUC @ 5 deg16.4
20
Visual LocalizationAachen Day-Night 1.0 (Night)
AUC @ (0.25m, 2°)86.7
18
Relative Pose EstimationMegaDepth-1800 (test)
Matches Count725
16
Visual LocalizationAachen Day-Night 1.0 (Day)
AUC (0.25m, 2°)86.8
14
Showing 10 of 18 rows

Other info

Follow for update