update: extractors

hloc/extractors/d2net.py

@@ -5,10 +5,10 @@ import torch
 
 from ..utils.base_model import BaseModel
 
-d2net_path = Path(__file__).parent / "../../third_party/d2net"
+d2net_path = Path(__file__).parent / "../../third_party"
 sys.path.append(str(d2net_path))
-from lib.model_test import D2Net as _D2Net
-from lib.pyramid import process_multiscale
+from d2net.lib.model_test import D2Net as _D2Net
+from d2net.lib.pyramid import process_multiscale
 
 class D2Net(BaseModel):
     default_conf = {

hloc/extractors/lanet.py

@@ -5,9 +5,9 @@ import torch
 
 from ..utils.base_model import BaseModel
 
-lanet_path = Path(__file__).parent / "../../third_party/lanet"
+lanet_path = Path(__file__).parent / "../../third_party"
 sys.path.append(str(lanet_path))
-from network_v0.model import PointModel
+from lanet.network_v0.model import PointModel
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 

hloc/extractors/rekd.py

@@ -5,9 +5,9 @@ import torch
 
 from ..utils.base_model import BaseModel
 
-rekd_path = Path(__file__).parent / "../../third_party/REKD"
+rekd_path = Path(__file__).parent / "../../third_party"
 sys.path.append(str(rekd_path))
-from training.model.REKD import REKD as REKD_
+from REKD.training.model.REKD import REKD as REKD_
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")