---
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](https://arxiv.org/abs/2502.17237)

### Qualitataive examples
Here are some examples of top-1 retrieved images from the SF-XL test set, which has 2.8M images as database.

![teaser](https://github.com/user-attachments/assets/a90b8d4c-ab53-4151-aacc-93493d583713)

### Usage
You can use the model with torch.hub, as simple as this
```
import torch
model = torch.hub.load("gmberton/MegaLoc", "get_trained_model")
```

### Gradio Demo
[Gradio Demo](https://bd40ca59030575ab1c.gradio.live)