Realcat commited on
Commit
44fac9f
·
1 Parent(s): d7ac6a3

format code

Browse files
hloc/extractors/alike.py CHANGED
@@ -3,7 +3,7 @@ from pathlib import Path
3
 
4
  import torch
5
 
6
- from hloc import logger, MODEL_REPO_ID
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 logger, MODEL_REPO_ID
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 logger, MODEL_REPO_ID
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 logger, MODEL_REPO_ID
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 logger, MODEL_REPO_ID
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 logger, MODEL_REPO_ID
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 logger, MODEL_REPO_ID
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 logger, MODEL_REPO_ID
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 logger, MODEL_REPO_ID
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 logger, MODEL_REPO_ID
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
- from hloc import MODEL_REPO_ID, DEVICE
 
 
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, DEVICE
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 logger, MODEL_REPO_ID
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 logger, MODEL_REPO_ID
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 logger, MODEL_REPO_ID
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, logger, MODEL_REPO_ID
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 logger, MODEL_REPO_ID
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 logger, MODEL_REPO_ID, DEVICE
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 logger, MODEL_REPO_ID
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 logger, MODEL_REPO_ID
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 logger, MODEL_REPO_ID
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 logger, MODEL_REPO_ID
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 logger, MODEL_REPO_ID
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"))