metadata
license: mit
tags:
- model_hub_mixin
- pytorch_model_hub_mixin
- arxiv:2502.17237
pipeline_tag: visual-document-retrieval
library_name: pytorch
MegaLoc
MegaLoc is an image retrieval model for any localization task, which achieves SOTA on most VPR datasets, including indoor and outdoor ones. You can find details in our paper MegaLoc: One Retrieval to Place Them All
Qualitataive examples
Here are some examples of top-1 retrieved images from the SF-XL test set, which has 2.8M images as database.
Usage
You can use the model with torch.hub, as simple as this
import torch
model = torch.hub.load("gmberton/MegaLoc", "get_trained_model")