图像匹配天花板:SuperPoint+SuperGlue复现

最近工作原因接触到图像匹配,经过调研发现SuperPoint+SuperGlue方法简直是图像匹配届的天花板,各种精度比较以及运行时间真令人惊讶,如下:
图像匹配天花板:SuperPoint+SuperGlue复现_第1张图片

图像匹配天花板:SuperPoint+SuperGlue复现_第2张图片
后来图像匹配(也可以做视频匹配,我这里做的是图像匹配)复现也特别简单,代码写得真好,环境配置好了也不怎么报错。下面直接来复现步骤,本人小白,如有不对,还请指出,互相交流,共同进步。
代码地址: https://github.com/magicleap/SuperGluePretrainedNetwork

  1. 下载以后解压,可以新建一个文件夹存储图像数据。
    图像匹配天花板:SuperPoint+SuperGlue复现_第3张图片
  2. 打开requirements文档查看并安装以下模块
    方法一:
    matplotlib>=3.1.3
    torch>=1.1.0
    opencv-python==4.1.2.30
    numpy>=1.18.1
    查看以后按需安装,适用于仅缺少个别的模块,或网速慢的情况下。
    方法二:
    cmd命令行,cd到该文件包路径下,激活要用的python虚拟环境,安装:
pip install -r requirements.txt
  1. 运行demo_superglue.py
    有两种方式。
    一是在cmd命令行输入:
 python demo_superglue.py --input = D:\SuperGlue\data --output_dir = D:\SuperGlue\data\result

–input是图像文件路径,必须输入,–output_dir是存储结果路径,如果不传入参数,则默认不进行保存。
其他还有很多参数,可以按需传入,打开demo_superglue.py可以查看。
图像匹配天花板:SuperPoint+SuperGlue复现_第4张图片
第二种方式,我称之为野路子,适合我这种小白。
打开demo_superglue.py,直接改default值,
图像匹配天花板:SuperPoint+SuperGlue复现_第5张图片

结束查看结果

图像匹配天花板:SuperPoint+SuperGlue复现_第6张图片

图像匹配天花板:SuperPoint+SuperGlue复现_第7张图片

你可能感兴趣的:(深度学习,图像特征匹配,代码复现,python,开发语言,经验分享)