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

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

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

  1. 下载以后解压,可以新建一个文件夹存储图像数据。
    在这里插入图片描述
  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可以查看。
在这里插入图片描述
第二种方式,我称之为野路子,适合我这种小白。
打开demo_superglue.py,直接改default值,
在这里插入图片描述

结束查看结果

在这里插入图片描述

在这里插入图片描述

  • 26
    点赞
  • 184
    收藏
    觉得还不错? 一键收藏
  • 41
    评论
SuperPointSuperGlue是一种基于深度学习的图像配准方法。SuperPoint用于提取图像中的关键点,而SuperGlue用于将两幅图像的关键点进行匹配和配准。 由于官方发布的SuperPointSuperGlue模型是基于COCO数据集训练的,可能与实际业务中的数据存在差距。因此,可以使用开源的pytorch-superpoint和pytorch-superglue来实现自己的图像配准任务。 具体步骤如下: 1. 安装pytorch-superpoint和pytorch-superglue库。 2. 使用pytorch-superpoint库提取图像中的关键点。可以参考该库的文档和示例代码来了解如何使用。 3. 使用pytorch-superglue库进行关键点的匹配和配准。该库提供了一些预训练的模型,可以直接使用或进行微调。 4. 根据实际需求,对配准结果进行后处理,例如去除错误匹配或优化配准结果。 下面是一个示例代码,演示了如何使用pytorch-superpoint和pytorch-superglue进行图像配准: ```python import torch import cv2 from superpoint import SuperPointFrontend from superglue import SuperGlueFrontend # 加载SuperPoint模型 superpoint = SuperPointFrontend(weights_path='superpoint.pth') # 加载SuperGlue模型 superglue = SuperGlueFrontend(weights_path='superglue.pth') # 读取两幅图像 image1 = cv2.imread('image1.jpg') image2 = cv2.imread('image2.jpg') # 提取关键点 points1, desc1, heatmap1 = superpoint.run(image1) points2, desc2, heatmap2 = superpoint.run(image2) # 进行关键点匹配和配准 matches, conf = superglue.run(image1, image2, points1, points2) # 可以根据需要对配准结果进行后处理 # 输出配准结果 print(matches) ``` 请注意,上述代码仅为示例,实际使用时需要根据具体情况进行适当的修改和调整。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 41
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值