ICLR 2024|图像匹配新突破!GIM:首个从互联网视频中学习通用图像匹配器的框架
前言 为了解决基于深度学习方法泛化性的问题,来自厦门大学、Intel、大疆的研究者们提出了 GIM: Learning Generalizable Image Matcher from Internet Videos。GIM 是第一个可以让匹配模型从互联网视频中学习到强泛化能力的训练框架。
本文转载自机器之心
仅用于学术分享,若侵权请联系删除
欢迎关注公众号CV技术指南,专注于计算机视觉的技术总结、最新技术跟踪、经典论文解读、CV招聘信息。
【CV技术指南】CV全栈指导班、基础入门班、论文指导班 全面上线!!
![](https://pic2.zhimg.com/80/v2-373062f504a3868bd8f9debb9c824fe5_1440w.webp)
论文主页:https://xuelunshen.com/gim论文地址:https://arxiv.org/abs/2402.11095论文视频:https://www.youtube.com/watch?v=FU_MJLD8LeY代码地址:https://github.com/xuelunshen/gim在线模型:https://huggingface.co/spaces/xuelunshen/gim-onlineGIM 从易于获取、种类繁多且近乎于无限的互联网视频 (如图 1 所示) 中提取有效的监督信号,用于匹配模型的训练。
![](https://pic1.zhimg.com/80/v2-54611869caf57ff808967a46f22b0808_1440w.webp)
图 1. 部分互联网视频包含的多样场景GIM 框架适用于训练所有匹配模型。如图 2 所示,三个匹配模型 DKM、LoFTR 和 SuperGlue 分别对应了:密集匹配、半密集匹配和稀疏匹配这三种主流匹配范式。在 GIM 框架下,用于训练的视频时长越长,匹配模型的性能就会越高。从目前的折线来看,在使用 50 小时的视频后,性能还没有表现出饱和的现象,因此如果使用更多的视频,性能还会进一步提高。
![](https://pic3.zhimg.com/80/v2-6c0d6f91c14b47a65b58d2695cd61006_1440w.webp)
图 2. 用于训练的视频时长和模型泛化性能的关系
为了能充分衡量一个匹配模型的泛化性能,作者提出了第一个 Zero-shot Evaluation Benchmark (ZEB)。如图 3 所示,ZEB 由 12 个涵盖各种场景、天气和相机模型的公开数据集组成,大约包含 4.6 万对测试图片。ZEB 所包含的数据数量远多于现有方法普遍采用的,由 MegaDepth 和 ScanNet 组成,仅包含 3000 对图像的测试集。
![](https://pic2.zhimg.com/80/v2-4e332befef0f833163e5454033d896b5_1440w.webp)
图 3.ZEB 的组成情况
图 2 中的模型泛化性能正是来自 ZEB 的评估结果,具体的评估结果在图 4 展示。Mean Rank 指标代表匹配模型在 12 个测试序列中的平均排名,排名越接近于 1 越好。Mean AUC@5° 代表匹配模型对姿态估计的准确度,数值越高越好。
![](https://pic1.zhimg.com/80/v2-efa18b7ae61b288016a7891dc6aaa790_1440w.webp)
图 4. 匹配模型泛化性能评估结果从图 4 可以看到,GIM 可以明显地提升匹配模型的泛化性能。在被 GIM 训练前,SuperGlue 和 LoFTR 在 GL3、BLE 和 GTA 序列中甚至不如传统算法 RootSIFT。在被 GIM 训练后,这两个模型的泛化性能都得到了极大的提升。正如刚才所说,50 小时的视频数据还远未让模型的性能达到极限。按照作者最新的 100 小时视频数据训练结果,性能已经可以达到 51.2,远超现有的模型。如图 5 所示,匹配模型的泛化性能提升,同样会为下游任务带来增益。值得注意的是,在图 5 的重定位任务中,GIM 都是仅以一个模型和其他特定场景的模型进行对比,但依然能取得更好的结果。
![](https://pic4.zhimg.com/80/v2-6047971a04eebec462bf5c7c4195cc07_1440w.webp)
图 5. 匹配模型在下游任务:单应性矩阵估计、室内和室外重定位任务下的性能表现双视图图像匹配的结果如下图所示:
![](https://pic4.zhimg.com/80/v2-a19568186f53afe96e263887ff3e39cb_1440w.webp)
图 6. 匹配模型在双视图匹配下的可视化结果除此之外,如图 7 所示,GIM 模型强大的泛化性能还可以处理训练中从未见过的点云鸟瞰图匹配任务。
![](https://pic4.zhimg.com/80/v2-a3db26d179638ef48be24a5f647118eb_1440w.webp)
图 7. 匹配模型在点云鸟瞰图匹配下的可视化结果作为多视图重建的通用方法,COLMAP 被广泛地应用于如 visual localization,neural rendering 等下游任务。将 COLMAP 中的匹配方法替换成 GIM,多视图重建的效果更好,这极大程度地提升了 COLMAP 的鲁棒性。
![](https://pic1.zhimg.com/80/v2-77b8b1b8fffc6bfc56627b35432ab628_1440w.webp)
图 8. 不同匹配模型对多视图重建的影响。第一行是部分重建图像的展示。第二行是重建的结果展示。读者感兴趣可以去论文主页和介绍视频中观看动态可交互的结果。由于 GIM 能够有效提升 COLMAP 多视角重建的质量,因此,GIM 也能为对应的下游任务中「增光添彩」,下面几张图展示了用 GIM 的 COLMAP 结果来初始化 Gaussian Splatting 的效果。可以看到在一些有挑战性的场景中,基于 SIFT 和 DKM 的 COLMAP 都不能够很好地提供有效的 initialization,导致渲染效果不佳。
![](https://pic2.zhimg.com/80/v2-4a7493ad64ae8fd58e778c1e79acb7e5_1440w.webp)
![](https://pic4.zhimg.com/80/v2-705fee5a2f505cdb12dcc50759083ebf_1440w.webp)
![](https://pic3.zhimg.com/80/v2-776bba21f9db705221618bccd32be0b2_1440w.webp)
图 9. 不同匹配模型在多视图重建后对 Gaussian Splatting 的影响。读者感兴趣可以去论文主页和介绍视频中观看动态结果。
框架方法
![](https://pic2.zhimg.com/80/v2-e3436a91c1a715aab983a3ead28f38dd_1440w.webp)
图 10.GIM 框架GIM 框架的方法很简洁,方法的核心在于利用视频中帧和帧之间的连续性,将匹配从短距离的帧传递到长距离的帧上,以此获取宽基线的训练图像。第一步,准备一个用于训练的匹配模型和互联网视频。第二步,用标准训练数据(非互联网视频,如 MegaDepth)训练匹配模型,然后再收集其他补充的匹配方法,用所有这些方法在互联网视频上的短距离间隔的帧上进行匹配,获取较为密集的匹配结果。再用 outlier filtering 过滤一遍匹配。第三步,利用视频的时序关系,将匹配传递到更远的帧上,得到重合度较低和基线较宽的训练图像对。最后,对传递后的图像对及其匹配标签进行数据增强并进行训练。下图是 GIM 在互联网视频上生成的匹配标签可视化。这仅仅是极小的一部分,用于训练的数据的多样性远远多于作者能展示出来的图片。
![](https://pic2.zhimg.com/80/v2-01d623f41977916b4e07825e93909bd9_1440w.webp)
图 11. 部分互联网视频标签的可视化总结GIM 的提出受到了 GPT 等大模型的启发,促使作者用海量的视频数据去训练匹配模型,并达到更高的泛化性能。GIM 的主页的地址是 http://xuelunshen.com/gim。欢迎大家跳转到 GIM 的主页,浏览更详细生动的介绍视频、开源的代码,在线体验 GIM 匹配效果的 HuggingFace Demo。
欢迎关注公众号CV技术指南,专注于计算机视觉的技术总结、最新技术跟踪、经典论文解读、CV招聘信息。
【技术文档】《从零搭建pytorch模型教程》122页PDF下载
QQ交流群:470899183。群内有大佬负责解答大家的日常学习、科研、代码问题。
其它文章
LSKA注意力 | 重新思考和设计大卷积核注意力,性能优于ConvNeXt、SWin、RepLKNet以及VAN
CVPR 2023 | TinyMIM:微软亚洲研究院用知识蒸馏改进小型ViT
ICCV2023|涨点神器!目标检测蒸馏学习新方法,浙大、海康威视等提出
ICCV 2023 Oral | 突破性图像融合与分割研究:全时多模态基准与多交互特征学习
HDRUNet | 深圳先进院董超团队提出带降噪与反量化功能的单帧HDR重建算法
南科大提出ORCTrack | 解决DeepSORT等跟踪方法的遮挡问题,即插即用真的很香
1800亿参数,世界顶级开源大模型Falcon官宣!碾压LLaMA 2,性能直逼GPT-4
SAM-Med2D:打破自然图像与医学图像的领域鸿沟,医疗版 SAM 开源了!
GhostSR|针对图像超分的特征冗余,华为诺亚&北大联合提出GhostSR
Meta推出像素级动作追踪模型,简易版在线可玩 | GitHub 1.4K星
CSUNet | 完美缝合Transformer和CNN,性能达到UNet家族的巅峰!