| 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. | |
|  | |
| ### 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) |