diff --git a/lightglue/__init__.py b/lightglue/__init__.py index b834c6d..5e94008 100644 --- a/lightglue/__init__.py +++ b/lightglue/__init__.py @@ -1,5 +1,5 @@ +from .aliked import ALIKED # noqa from .disk import DISK # noqa from .lightglue import LightGlue # noqa from .superpoint import SuperPoint # noqa -from .aliked import ALIKED # noqa from .utils import match_pair # noqa diff --git a/lightglue/aliked.py b/lightglue/aliked.py index 96ba36f..6c6e593 100644 --- a/lightglue/aliked.py +++ b/lightglue/aliked.py @@ -32,8 +32,8 @@ # Xiaoming Zhao, Xingming Wu, Weihai Chen, Peter C.Y. Chen, Qingsong Xu, and Zhengguo Li # Code from https://github.com/Shiaoming/ALIKED -from typing import Callable, Optional from types import SimpleNamespace +from typing import Callable, Optional import torch import torch.nn.functional as F diff --git a/lightglue/sift.py b/lightglue/sift.py index 6138b56..b9c96b1 100644 --- a/lightglue/sift.py +++ b/lightglue/sift.py @@ -1,3 +1,5 @@ +from types import SimpleNamespace + import cv2 import numpy as np import pycolmap @@ -5,7 +7,6 @@ import torch.nn as nn from omegaconf import OmegaConf from scipy.spatial import KDTree -from types import SimpleNamespace from .utils import ImagePreprocessor diff --git a/lightglue/sift_kornia.py b/lightglue/sift_kornia.py index 18feb80..a9d41b1 100644 --- a/lightglue/sift_kornia.py +++ b/lightglue/sift_kornia.py @@ -1,9 +1,9 @@ +from types import SimpleNamespace + import kornia import torch import torch.nn as nn -from types import SimpleNamespace - from .utils import ImagePreprocessor