magicleap / SuperGluePretrainedNetwork

SuperGlue: Learning Feature Matching with Graph Neural Networks (CVPR 2020, Oral)
Other
3.31k stars 672 forks source link

About the sinkhorn implementation #100

Open bluenights opened 3 years ago

bluenights commented 3 years ago

Thanks for the code! I'm wondering where the entropy regulation parameter is implemented in the code.

sarlinpe commented 2 years ago

Sorry for the late reply. The matching descriptors are not normalized so the network has control over the scale of the Sinkhorn scores. This is redundant with the temperature parameter of the original Sinkhorn formulation, so we removed it. During the course of training, the scores are increasing in magnitude as the network gains confidence in its predictions.