Weatherproofing Retrieval for Localization with Generative AI and Geometric Consistency
Authors: Yannis Kalantidis,Mert Bülent Sarıyıldız,Rafael S. Rezende,Philippe Weinzaepfel,Diane Larlus,Gabriela Csurka
Presentation: ICLR 2024,Poster

Abstract: State-of-the-art visual localization approaches generally rely on a first image retrieval step whose role is crucial. Yet, retrieval often struggles when facing varying conditions, due to e.g. weather or time of day, with dramatic consequences on the visual localization accuracy. In this paper, we improve this retrieval step and tailor it to the final localization task. Among the several changes we advocate for, we propose to synthesize variants of the training set images, obtained from generative text-to-image models, in order to automatically expand the training set towards a number of nameable variations that particularly hurt visual localization. After expanding the training set, we propose a training approach that leverages the specificities and the underlying geometry of this mix of real and synthetic images. We experimentally show that those changes translate into large improvements for the most challenging visual localization datasets.