文章

快速了解一个网络:Super Glue, Learning Feature Matching with Graph Neural Networks

快速了解一个网络:Super Glue, Learning Feature Matching with Graph Neural Networks

以下内容偏向于记录个人学习过程及思考,请审慎阅读。

背景

对于两张图像的特征匹配:较大的视点和光照变化、遮挡、模糊和缺乏纹理是使 2D 到 2D 数据关联具有较大的挑战性。

核心思想

Super Glue 从特征点匹配角度(而非特征点提取角度)出发,通过Graph Neural Networks (GNN)建模特征点间的匹配关系。

Pipeline

super-glue-pipeline

亿些细节

Multiplex Graph Neural Network

对于第l层、图A中的第i个特征点,通过GNN结构将其与图A中的其它特征点及图B中的特征点做交互,最终得到汇聚的特征与当前层输入特征进行concat,经过MLP后再与当前层输入特征相加。

Attentional Aggregation

每层汇聚的特征是通过attention机制得到的。

  • 对于奇数层,汇聚的是本图中特征点交互的特征;
  • 对于偶数层,汇聚的是它图中特征点交互的特征;

super-glue-attention

Score Prediction

在最优匹配过程中,特征间的相似度是通过特征向量的内积得到的。

Occlusion and Visibility

对于各图中没有匹配上的点,会将这些点都指定其匹配到一个dustbin(垃圾桶)上,以使网络抑制某些特征点。

Sinkhorn Algorithm

最优匹配算法采用Sinkhorn Algorithm,是一种可微的Hungarian Algorithm

Loss

分配方案的负对数似然,并且包含未匹配上的点。

进一步了解

原文和代码

https://arxiv.org/abs/1911.11763

参考资料

论文阅读《Super Glue, Learning Feature Matching with Graph Neural Networks》

本文由作者按照 CC BY 4.0 进行授权