fix: match dense

Vincentqyw · Oct 30, 2024 · fbaa554 · fbaa554
1 parent 08fa558
commit fbaa554
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/hloc/match_dense.py b/hloc/match_dense.py
@@ -250,6 +250,7 @@
         "model": {
             "name": "roma",
             "weights": "outdoor",
+            "model_name": "roma_outdoor.pth",
             "max_keypoints": 2000,
             "match_threshold": 0.2,
         },
@@ -266,7 +267,7 @@
         "output": "matches-gim",
         "model": {
             "name": "gim",
-            "weights": "gim_dkm_100h.ckpt",
+            "model_name": "gim_dkm_100h.ckpt",
             "max_keypoints": 2000,
             "match_threshold": 0.2,
         },