/MNNSuperGlue

SuperGlue MNN C++部署，SuperGlue C++ Inference with MNN

Primary LanguageC++

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
3.cd build
4.cmake ../
5.make

  • 执行

    ./build/kptsdet

后续

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

致谢

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