Gabriele
commited on
Commit
·
13ec015
1
Parent(s):
53edac2
Remove PyTorchModelHubMixin dependency
Browse files- README.md +0 -1
- config.json +0 -7
- megaloc_model.py +1 -2
README.md
CHANGED
|
@@ -5,7 +5,6 @@ license: mit
|
|
| 5 |
tags:
|
| 6 |
- visual-place-recognition
|
| 7 |
- image-retrieval
|
| 8 |
-
- pytorch_model_hub_mixin
|
| 9 |
- arxiv:2502.17237
|
| 10 |
---
|
| 11 |
|
|
|
|
| 5 |
tags:
|
| 6 |
- visual-place-recognition
|
| 7 |
- image-retrieval
|
|
|
|
| 8 |
- arxiv:2502.17237
|
| 9 |
---
|
| 10 |
|
config.json
DELETED
|
@@ -1,7 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"feat_dim": 8448,
|
| 3 |
-
"num_clusters": 64,
|
| 4 |
-
"cluster_dim": 256,
|
| 5 |
-
"token_dim": 256,
|
| 6 |
-
"mlp_dim": 512
|
| 7 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
megaloc_model.py
CHANGED
|
@@ -15,7 +15,6 @@ import torch
|
|
| 15 |
import torch.nn as nn
|
| 16 |
import torch.nn.functional as F
|
| 17 |
import torchvision.transforms.functional as tfm
|
| 18 |
-
from huggingface_hub import PyTorchModelHubMixin
|
| 19 |
|
| 20 |
|
| 21 |
# Code adapted from OpenGlue, MIT license
|
|
@@ -404,7 +403,7 @@ class Aggregator(nn.Module):
|
|
| 404 |
return self.linear(x)
|
| 405 |
|
| 406 |
|
| 407 |
-
class MegaLoc(nn.Module
|
| 408 |
"""MegaLoc: Unified visual place recognition model.
|
| 409 |
|
| 410 |
Combines a DINOv2 Vision Transformer backbone with optimal transport-based
|
|
|
|
| 15 |
import torch.nn as nn
|
| 16 |
import torch.nn.functional as F
|
| 17 |
import torchvision.transforms.functional as tfm
|
|
|
|
| 18 |
|
| 19 |
|
| 20 |
# Code adapted from OpenGlue, MIT license
|
|
|
|
| 403 |
return self.linear(x)
|
| 404 |
|
| 405 |
|
| 406 |
+
class MegaLoc(nn.Module):
|
| 407 |
"""MegaLoc: Unified visual place recognition model.
|
| 408 |
|
| 409 |
Combines a DINOv2 Vision Transformer backbone with optimal transport-based
|