4 个月前

神经引导的RANSAC:学习采样模型假设的位置

神经引导的RANSAC:学习采样模型假设的位置

摘要

我们介绍了神经引导随机样本一致性算法(Neural-Guided RANSAC,简称NG-RANSAC),这是对经典鲁棒优化算法RANSAC的一种扩展。NG-RANSAC利用先验信息来改进模型假设搜索,从而增加找到无离群点最小集的可能性。以往的研究通常使用启发式辅助信息(如手工设计的描述符距离)来指导假设搜索。相比之下,我们以一种系统的方式学习假设搜索,这使得我们在训练过程中可以优化任意任务损失,从而在经典的计算机视觉任务上取得显著的改进。我们进一步提出了NG-RANSAC的两个扩展:首先,使用内点计数本身作为训练信号,使我们能够以自监督的方式训练神经引导;其次,我们将神经引导与可微分RANSAC结合,构建出专注于输入数据特定部分并尽可能优化输出预测的神经网络。我们在一系列广泛的计算机视觉任务上评估了NG-RANSAC,包括极线几何估计、地平线估计和相机重定位。实验结果表明,与最先进的鲁棒估计器(包括非常近期的学习型估计器)相比,我们的方法取得了优越或具有竞争力的结果。

代码仓库

vislearn/ngdsac_camreloc
pytorch
GitHub 中提及
vislearn/ngdsac_horizon
pytorch
GitHub 中提及
vislearn/ngransac
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
horizon-line-estimation-on-horizon-lines-inNG-DSAC
AUC (horizon error): 75.2

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
神经引导的RANSAC:学习采样模型假设的位置 | 论文 | HyperAI超神经