MNNSuperGlue

概述

MNN Superglue 关键点匹配C++实现，原论文《SuperGlue: Learning Feature Matching with Graph Neural Networks (CVPR 2020, Oral)》，原pytorch代码https://github.com/magicleap/SuperGluePretrainedNetwork

编译执行

• Cmake & make

  1.修改Cmake里面MNN库的路径，按照自己实际路径修改
  2.mkdir build
  1.cd build
  2.cmake ../
  3.make
  

• 执行

  ./build/kptsdet
  

后续

由于原版MNN有些算子不支持CUDA后端，所以目前只在CPU端测试通过，后续有时间会尝试写CUDA后端相关的算子。如果有兴趣的小伙伴也可以一起完善。

致谢

本仓库代码都是翻译https://github.com/magicleap/SuperGluePretrainedNetwork中pytorch代码，具体细节请参考原pytorch版，原理部分可以阅读论文《https://arxiv.org/pdf/1712.07629.pdf》