Skip to content

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

Notifications You must be signed in to change notification settings

Hanson0910/MNNSuperGlue

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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》

About

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published