The task of Visual Place Recognition (VPR) aims to match a query image against references from an extensive database of images from different places, relying solely on visual cues. State-of-the-art pipelines focus on the aggregation of features extracted from a deep backbone, in order to form a global descriptor for each image. In this context, we introduce SALAD (Sinkhorn Algorithm for Locally Aggregated Descriptors), which reformulates NetVLAD's soft-assignment of local features to clusters as an optimal transport problem. In SALAD, we consider both feature-to-cluster and cluster-to-feature relations and we also introduce a 'dustbin' cluster, designed to selectively discard features deemed non-informative, enhancing the overall descriptor quality. Additionally, we leverage and fine-tune DINOv2 as a backbone, which provides enhanced description power for the local features, and dramatically reduces the required training time. As a result, our single-stage method not only surpasses single-stage baselines in public VPR datasets, but also surpasses two-stage methods that add a re-ranking with significantly higher cost. Code and models are available at https://github.com/serizba/salad.
翻译:视觉地点识别(VPR)的任务旨在仅依靠视觉线索,将查询图像与来自不同地点的庞大图像数据库中的参考图像进行匹配。先进的流程侧重于聚合从深度骨干网络提取的特征,以形成每幅图像的全局描述符。在此背景下,我们提出了SALAD(基于Sinkhorn算法的局部聚合描述符),该方法将NetVLAD中局部特征到聚类的软分配重新表述为最优传输问题。在SALAD中,我们同时考虑了特征到聚类和聚类到特征的关系,并引入了一个“垃圾箱”聚类,旨在选择性丢弃被认为无信息的特征,从而提升描述符的整体质量。此外,我们利用并微调了DINOv2作为骨干网络,它为局部特征提供了更强的描述能力,并显著减少了所需的训练时间。因此,我们的单阶段方法不仅在公开VPR数据集上超越了单阶段基线,还超越了添加重排序(成本显著更高)的两阶段方法。代码和模型可在https://github.com/serizba/salad获取。