Why use argmax for matching?

[gabrielhuang]

Hey there, thanks for sharing the code!

I just have a quick question:
Why is the argmax used to match features from different views, rather than using the spatial correspondance, which we have access to, since we know which data augmentations were applied to the images?

Thanks.