Realcat
commited on
Commit
·
44fac9f
1
Parent(s):
d7ac6a3
format code
Browse files- hloc/extractors/alike.py +1 -1
- hloc/extractors/d2net.py +1 -1
- hloc/extractors/darkfeat.py +1 -1
- hloc/extractors/dedode.py +1 -1
- hloc/extractors/lanet.py +1 -1
- hloc/extractors/r2d2.py +1 -1
- hloc/extractors/rekd.py +1 -1
- hloc/extractors/rord.py +1 -1
- hloc/extractors/sfd2.py +1 -1
- hloc/matchers/aspanformer.py +1 -1
- hloc/matchers/cotr.py +3 -1
- hloc/matchers/dkm.py +1 -2
- hloc/matchers/duster.py +1 -1
- hloc/matchers/eloftr.py +1 -0
- hloc/matchers/gim.py +2 -1
- hloc/matchers/gluestick.py +1 -1
- hloc/matchers/imp.py +1 -1
- hloc/matchers/lightglue.py +1 -1
- hloc/matchers/mast3r.py +1 -1
- hloc/matchers/mickey.py +1 -1
- hloc/matchers/omniglue.py +1 -1
- hloc/matchers/roma.py +1 -1
- hloc/matchers/sgmnet.py +1 -1
- hloc/matchers/sold2.py +1 -1
- hloc/matchers/topicfm.py +2 -0
hloc/extractors/alike.py
CHANGED
@@ -3,7 +3,7 @@ from pathlib import Path
|
|
3 |
|
4 |
import torch
|
5 |
|
6 |
-
from hloc import
|
7 |
|
8 |
from ..utils.base_model import BaseModel
|
9 |
|
|
|
3 |
|
4 |
import torch
|
5 |
|
6 |
+
from hloc import MODEL_REPO_ID, logger
|
7 |
|
8 |
from ..utils.base_model import BaseModel
|
9 |
|
hloc/extractors/d2net.py
CHANGED
@@ -3,7 +3,7 @@ from pathlib import Path
|
|
3 |
|
4 |
import torch
|
5 |
|
6 |
-
from hloc import
|
7 |
|
8 |
from ..utils.base_model import BaseModel
|
9 |
|
|
|
3 |
|
4 |
import torch
|
5 |
|
6 |
+
from hloc import MODEL_REPO_ID, logger
|
7 |
|
8 |
from ..utils.base_model import BaseModel
|
9 |
|
hloc/extractors/darkfeat.py
CHANGED
@@ -2,7 +2,7 @@ import subprocess
|
|
2 |
import sys
|
3 |
from pathlib import Path
|
4 |
|
5 |
-
from hloc import
|
6 |
|
7 |
from ..utils.base_model import BaseModel
|
8 |
|
|
|
2 |
import sys
|
3 |
from pathlib import Path
|
4 |
|
5 |
+
from hloc import MODEL_REPO_ID, logger
|
6 |
|
7 |
from ..utils.base_model import BaseModel
|
8 |
|
hloc/extractors/dedode.py
CHANGED
@@ -4,7 +4,7 @@ from pathlib import Path
|
|
4 |
import torch
|
5 |
import torchvision.transforms as transforms
|
6 |
|
7 |
-
from hloc import
|
8 |
|
9 |
from ..utils.base_model import BaseModel
|
10 |
|
|
|
4 |
import torch
|
5 |
import torchvision.transforms as transforms
|
6 |
|
7 |
+
from hloc import MODEL_REPO_ID, logger
|
8 |
|
9 |
from ..utils.base_model import BaseModel
|
10 |
|
hloc/extractors/lanet.py
CHANGED
@@ -3,7 +3,7 @@ from pathlib import Path
|
|
3 |
|
4 |
import torch
|
5 |
|
6 |
-
from hloc import
|
7 |
|
8 |
from ..utils.base_model import BaseModel
|
9 |
|
|
|
3 |
|
4 |
import torch
|
5 |
|
6 |
+
from hloc import MODEL_REPO_ID, logger
|
7 |
|
8 |
from ..utils.base_model import BaseModel
|
9 |
|
hloc/extractors/r2d2.py
CHANGED
@@ -3,7 +3,7 @@ from pathlib import Path
|
|
3 |
|
4 |
import torchvision.transforms as tvf
|
5 |
|
6 |
-
from hloc import
|
7 |
|
8 |
from ..utils.base_model import BaseModel
|
9 |
|
|
|
3 |
|
4 |
import torchvision.transforms as tvf
|
5 |
|
6 |
+
from hloc import MODEL_REPO_ID, logger
|
7 |
|
8 |
from ..utils.base_model import BaseModel
|
9 |
|
hloc/extractors/rekd.py
CHANGED
@@ -3,7 +3,7 @@ from pathlib import Path
|
|
3 |
|
4 |
import torch
|
5 |
|
6 |
-
from hloc import
|
7 |
|
8 |
from ..utils.base_model import BaseModel
|
9 |
|
|
|
3 |
|
4 |
import torch
|
5 |
|
6 |
+
from hloc import MODEL_REPO_ID, logger
|
7 |
|
8 |
from ..utils.base_model import BaseModel
|
9 |
|
hloc/extractors/rord.py
CHANGED
@@ -4,7 +4,7 @@ from pathlib import Path
|
|
4 |
|
5 |
import torch
|
6 |
|
7 |
-
from hloc import
|
8 |
|
9 |
from ..utils.base_model import BaseModel
|
10 |
|
|
|
4 |
|
5 |
import torch
|
6 |
|
7 |
+
from hloc import MODEL_REPO_ID, logger
|
8 |
|
9 |
from ..utils.base_model import BaseModel
|
10 |
|
hloc/extractors/sfd2.py
CHANGED
@@ -3,7 +3,7 @@ from pathlib import Path
|
|
3 |
|
4 |
import torchvision.transforms as tvf
|
5 |
|
6 |
-
from .. import
|
7 |
from ..utils.base_model import BaseModel
|
8 |
|
9 |
tp_path = Path(__file__).parent / "../../third_party"
|
|
|
3 |
|
4 |
import torchvision.transforms as tvf
|
5 |
|
6 |
+
from .. import MODEL_REPO_ID, logger
|
7 |
from ..utils.base_model import BaseModel
|
8 |
|
9 |
tp_path = Path(__file__).parent / "../../third_party"
|
hloc/matchers/aspanformer.py
CHANGED
@@ -4,7 +4,7 @@ from pathlib import Path
|
|
4 |
|
5 |
import torch
|
6 |
|
7 |
-
from hloc import
|
8 |
from hloc.utils.base_model import BaseModel
|
9 |
|
10 |
sys.path.append(str(Path(__file__).parent / "../../third_party"))
|
|
|
4 |
|
5 |
import torch
|
6 |
|
7 |
+
from hloc import MODEL_REPO_ID, logger
|
8 |
from hloc.utils.base_model import BaseModel
|
9 |
|
10 |
sys.path.append(str(Path(__file__).parent / "../../third_party"))
|
hloc/matchers/cotr.py
CHANGED
@@ -5,7 +5,9 @@ from pathlib import Path
|
|
5 |
import numpy as np
|
6 |
import torch
|
7 |
from torchvision.transforms import ToPILImage
|
8 |
-
|
|
|
|
|
9 |
from ..utils.base_model import BaseModel
|
10 |
|
11 |
sys.path.append(str(Path(__file__).parent / "../../third_party/COTR"))
|
|
|
5 |
import numpy as np
|
6 |
import torch
|
7 |
from torchvision.transforms import ToPILImage
|
8 |
+
|
9 |
+
from hloc import DEVICE, MODEL_REPO_ID
|
10 |
+
|
11 |
from ..utils.base_model import BaseModel
|
12 |
|
13 |
sys.path.append(str(Path(__file__).parent / "../../third_party/COTR"))
|
hloc/matchers/dkm.py
CHANGED
@@ -4,8 +4,7 @@ from pathlib import Path
|
|
4 |
import torch
|
5 |
from PIL import Image
|
6 |
|
7 |
-
from hloc import MODEL_REPO_ID,
|
8 |
-
from hloc import logger
|
9 |
from hloc.utils.base_model import BaseModel
|
10 |
|
11 |
sys.path.append(str(Path(__file__).parent / "../../third_party"))
|
|
|
4 |
import torch
|
5 |
from PIL import Image
|
6 |
|
7 |
+
from hloc import DEVICE, MODEL_REPO_ID, logger
|
|
|
8 |
from hloc.utils.base_model import BaseModel
|
9 |
|
10 |
sys.path.append(str(Path(__file__).parent / "../../third_party"))
|
hloc/matchers/duster.py
CHANGED
@@ -5,7 +5,7 @@ import numpy as np
|
|
5 |
import torch
|
6 |
import torchvision.transforms as tfm
|
7 |
|
8 |
-
from .. import
|
9 |
from ..utils.base_model import BaseModel
|
10 |
|
11 |
duster_path = Path(__file__).parent / "../../third_party/dust3r"
|
|
|
5 |
import torch
|
6 |
import torchvision.transforms as tfm
|
7 |
|
8 |
+
from .. import MODEL_REPO_ID, logger
|
9 |
from ..utils.base_model import BaseModel
|
10 |
|
11 |
duster_path = Path(__file__).parent / "../../third_party/dust3r"
|
hloc/matchers/eloftr.py
CHANGED
@@ -4,6 +4,7 @@ from copy import deepcopy
|
|
4 |
from pathlib import Path
|
5 |
|
6 |
import torch
|
|
|
7 |
from hloc import MODEL_REPO_ID
|
8 |
|
9 |
tp_path = Path(__file__).parent / "../../third_party"
|
|
|
4 |
from pathlib import Path
|
5 |
|
6 |
import torch
|
7 |
+
|
8 |
from hloc import MODEL_REPO_ID
|
9 |
|
10 |
tp_path = Path(__file__).parent / "../../third_party"
|
hloc/matchers/gim.py
CHANGED
@@ -1,8 +1,9 @@
|
|
1 |
import sys
|
2 |
from pathlib import Path
|
|
|
3 |
import torch
|
4 |
|
5 |
-
from .. import
|
6 |
from ..utils.base_model import BaseModel
|
7 |
|
8 |
gim_path = Path(__file__).parent / "../../third_party/gim"
|
|
|
1 |
import sys
|
2 |
from pathlib import Path
|
3 |
+
|
4 |
import torch
|
5 |
|
6 |
+
from .. import MODEL_REPO_ID, logger
|
7 |
from ..utils.base_model import BaseModel
|
8 |
|
9 |
gim_path = Path(__file__).parent / "../../third_party/gim"
|
hloc/matchers/gluestick.py
CHANGED
@@ -3,7 +3,7 @@ from pathlib import Path
|
|
3 |
|
4 |
import torch
|
5 |
|
6 |
-
from .. import
|
7 |
from ..utils.base_model import BaseModel
|
8 |
|
9 |
gluestick_path = Path(__file__).parent / "../../third_party/GlueStick"
|
|
|
3 |
|
4 |
import torch
|
5 |
|
6 |
+
from .. import MODEL_REPO_ID, logger
|
7 |
from ..utils.base_model import BaseModel
|
8 |
|
9 |
gluestick_path = Path(__file__).parent / "../../third_party/GlueStick"
|
hloc/matchers/imp.py
CHANGED
@@ -3,7 +3,7 @@ from pathlib import Path
|
|
3 |
|
4 |
import torch
|
5 |
|
6 |
-
from .. import DEVICE,
|
7 |
from ..utils.base_model import BaseModel
|
8 |
|
9 |
tp_path = Path(__file__).parent / "../../third_party"
|
|
|
3 |
|
4 |
import torch
|
5 |
|
6 |
+
from .. import DEVICE, MODEL_REPO_ID, logger
|
7 |
from ..utils.base_model import BaseModel
|
8 |
|
9 |
tp_path = Path(__file__).parent / "../../third_party"
|
hloc/matchers/lightglue.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import sys
|
2 |
from pathlib import Path
|
3 |
|
4 |
-
from .. import
|
5 |
from ..utils.base_model import BaseModel
|
6 |
|
7 |
lightglue_path = Path(__file__).parent / "../../third_party/LightGlue"
|
|
|
1 |
import sys
|
2 |
from pathlib import Path
|
3 |
|
4 |
+
from .. import MODEL_REPO_ID, logger
|
5 |
from ..utils.base_model import BaseModel
|
6 |
|
7 |
lightglue_path = Path(__file__).parent / "../../third_party/LightGlue"
|
hloc/matchers/mast3r.py
CHANGED
@@ -5,7 +5,7 @@ import numpy as np
|
|
5 |
import torch
|
6 |
import torchvision.transforms as tfm
|
7 |
|
8 |
-
from .. import
|
9 |
|
10 |
mast3r_path = Path(__file__).parent / "../../third_party/mast3r"
|
11 |
sys.path.append(str(mast3r_path))
|
|
|
5 |
import torch
|
6 |
import torchvision.transforms as tfm
|
7 |
|
8 |
+
from .. import DEVICE, MODEL_REPO_ID, logger
|
9 |
|
10 |
mast3r_path = Path(__file__).parent / "../../third_party/mast3r"
|
11 |
sys.path.append(str(mast3r_path))
|
hloc/matchers/mickey.py
CHANGED
@@ -3,7 +3,7 @@ from pathlib import Path
|
|
3 |
|
4 |
import torch
|
5 |
|
6 |
-
from .. import
|
7 |
from ..utils.base_model import BaseModel
|
8 |
|
9 |
mickey_path = Path(__file__).parent / "../../third_party"
|
|
|
3 |
|
4 |
import torch
|
5 |
|
6 |
+
from .. import MODEL_REPO_ID, logger
|
7 |
from ..utils.base_model import BaseModel
|
8 |
|
9 |
mickey_path = Path(__file__).parent / "../../third_party"
|
hloc/matchers/omniglue.py
CHANGED
@@ -5,7 +5,7 @@ from pathlib import Path
|
|
5 |
import numpy as np
|
6 |
import torch
|
7 |
|
8 |
-
from .. import
|
9 |
from ..utils.base_model import BaseModel
|
10 |
|
11 |
thirdparty_path = Path(__file__).parent / "../../third_party"
|
|
|
5 |
import numpy as np
|
6 |
import torch
|
7 |
|
8 |
+
from .. import MODEL_REPO_ID, logger
|
9 |
from ..utils.base_model import BaseModel
|
10 |
|
11 |
thirdparty_path = Path(__file__).parent / "../../third_party"
|
hloc/matchers/roma.py
CHANGED
@@ -4,7 +4,7 @@ from pathlib import Path
|
|
4 |
import torch
|
5 |
from PIL import Image
|
6 |
|
7 |
-
from .. import
|
8 |
from ..utils.base_model import BaseModel
|
9 |
|
10 |
roma_path = Path(__file__).parent / "../../third_party/RoMa"
|
|
|
4 |
import torch
|
5 |
from PIL import Image
|
6 |
|
7 |
+
from .. import MODEL_REPO_ID, logger
|
8 |
from ..utils.base_model import BaseModel
|
9 |
|
10 |
roma_path = Path(__file__).parent / "../../third_party/RoMa"
|
hloc/matchers/sgmnet.py
CHANGED
@@ -4,7 +4,7 @@ from pathlib import Path
|
|
4 |
|
5 |
import torch
|
6 |
|
7 |
-
from .. import
|
8 |
from ..utils.base_model import BaseModel
|
9 |
|
10 |
sgmnet_path = Path(__file__).parent / "../../third_party/SGMNet"
|
|
|
4 |
|
5 |
import torch
|
6 |
|
7 |
+
from .. import MODEL_REPO_ID, logger
|
8 |
from ..utils.base_model import BaseModel
|
9 |
|
10 |
sgmnet_path = Path(__file__).parent / "../../third_party/SGMNet"
|
hloc/matchers/sold2.py
CHANGED
@@ -3,7 +3,7 @@ from pathlib import Path
|
|
3 |
|
4 |
import torch
|
5 |
|
6 |
-
from .. import
|
7 |
from ..utils.base_model import BaseModel
|
8 |
|
9 |
sold2_path = Path(__file__).parent / "../../third_party/SOLD2"
|
|
|
3 |
|
4 |
import torch
|
5 |
|
6 |
+
from .. import MODEL_REPO_ID, logger
|
7 |
from ..utils.base_model import BaseModel
|
8 |
|
9 |
sold2_path = Path(__file__).parent / "../../third_party/SOLD2"
|
hloc/matchers/topicfm.py
CHANGED
@@ -2,7 +2,9 @@ import sys
|
|
2 |
from pathlib import Path
|
3 |
|
4 |
import torch
|
|
|
5 |
from hloc import MODEL_REPO_ID
|
|
|
6 |
from ..utils.base_model import BaseModel
|
7 |
|
8 |
sys.path.append(str(Path(__file__).parent / "../../third_party"))
|
|
|
2 |
from pathlib import Path
|
3 |
|
4 |
import torch
|
5 |
+
|
6 |
from hloc import MODEL_REPO_ID
|
7 |
+
|
8 |
from ..utils.base_model import BaseModel
|
9 |
|
10 |
sys.path.append(str(Path(__file__).parent / "../../third_party"))
|