4 个月前

DISK:使用策略梯度学习局部特征

DISK:使用策略梯度学习局部特征

摘要

局部特征框架在端到端学习中存在困难,主要是由于稀疏关键点的选择和匹配过程中固有的离散性。我们提出了一种名为DISK(离散关键点)的新方法,该方法通过利用强化学习(Reinforcement Learning, RL)的原则,优化了端到端学习过程以实现高数量的正确特征匹配。我们的简单而富有表现力的概率模型使我们能够在保持训练和推理机制接近的同时,具备良好的收敛特性,从而可靠地从零开始训练。我们的特征可以非常密集地提取,同时仍具有辨别能力,这挑战了关于什么是良好关键点的普遍假设,如图1所示,并在三个公开基准测试中取得了最先进的结果。

代码仓库

cvlab-epfl/disk
官方
pytorch
GitHub 中提及
fabio-sim/LightGlue-ONNX
pytorch
GitHub 中提及

基准测试

基准方法指标
image-matching-on-imc-phototourismDISK
mean average accuracy @ 10: 0.65435

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
DISK:使用策略梯度学习局部特征 | 论文 | HyperAI超神经