4 个月前

迭代几何编码体积用于立体匹配

迭代几何编码体积用于立体匹配

摘要

循环全对场变换(RAFT)在匹配任务中展现出了巨大的潜力。然而,全对相关性缺乏非局部几何知识,并且在病态区域难以解决局部模糊问题。本文提出了一种新的立体匹配深度网络架构——迭代几何编码体(IGEV-Stereo)。所提出的 IGEV-Stereo 构建了一个组合几何编码体,该编码体不仅包含了几何和上下文信息,还编码了局部匹配细节,并通过迭代索引不断更新视差图。为了加速收敛,我们利用 GEV 回归一个准确的起始点,以便 ConvGRUs 迭代使用。我们的 IGEV-Stereo 在所有已发表的方法中,在 KITTI 2015 和 2012(反射)数据集上均排名第一,并且在前十名方法中速度最快。此外,IGEV-Stereo 具有强大的跨数据集泛化能力和高效的推理效率。我们还将 IGEV 扩展到多视图立体匹配(MVS),即 IGEV-MVS,在 DTU 基准测试中达到了具有竞争力的精度。代码可在 https://github.com/gangweiX/IGEV 获取。

代码仓库

gangweix/igev
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
omnnidirectional-stereo-depth-estimation-onIGEV-Stereo
Depth-LRCE: 1.203
Depth-MAE: 1.860
Depth-MARE: 0.146
Depth-RMSE: 4.474
Disp-MAE: 0.225
Disp-MARE: 0.172
Disp-RMSE: 0.423

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
迭代几何编码体积用于立体匹配 | 论文 | HyperAI超神经