new file: config/aliked+lightglue.yaml

new file: config/disk+lightglue.yaml new file: config/keynet+kornia_matcher.yaml modified: config/superpoint+lightglue.yaml modified: config/superpoint+superglue.yaml modified: src/deep_image_matching/config.py
nneilsutherland · May 29, 2024 · 84a7546 · 84a7546
1 parent 0534d61
commit 84a7546
Show file tree

Hide file tree

Showing 6 changed files with 64 additions and 2 deletions.
diff --git a/config/aliked+lightglue.yaml b/config/aliked+lightglue.yaml
@@ -0,0 +1,22 @@
+# User configuration file
+
+general:
+  tile_size: (2400, 2000)
+  geom_verification: pydegensac
+  gv_threshold: 1
+  min_inliers_per_pair: 10
+  min_inlier_ratio_per_pair: 0.25
+
+extractor:
+  name: "aliked"
+  max_num_keypoints: 8000 
+  detection_threshold: 0.2
+  nms_radius: 3
+
+matcher:
+  name: "lightglue"
+  flash: True # enable FlashAttention if available
+  mp: False # enable mixed precision
+  depth_confidence: 0.95 # early stopping, disable with -1
+  width_confidence: 0.99 # point pruning, disable with -1
+  filter_threshold: 0.10 # match threshold
diff --git a/config/disk+lightglue.yaml b/config/disk+lightglue.yaml
@@ -0,0 +1,20 @@
+# User configuration file
+
+general:
+  tile_size: (2400, 2000)
+  geom_verification: pydegensac
+  gv_threshold: 1
+  min_inliers_per_pair: 10
+  min_inlier_ratio_per_pair: 0.25
+
+extractor:
+  name: "disk"
+  max_keypoints: 8000
+
+matcher:
+  name: "lightglue"
+  flash: True # enable FlashAttention if available
+  mp: False # enable mixed precision
+  depth_confidence: 0.95 # early stopping, disable with -1
+  width_confidence: 0.99 # point pruning, disable with -1
+  filter_threshold: 0.10 # match threshold
diff --git a/config/keynet+kornia_matcher.yaml b/config/keynet+kornia_matcher.yaml
@@ -0,0 +1,18 @@
+# User configuration file
+
+general:
+  tile_size: (2400, 2000)
+  geom_verification: pydegensac
+  gv_threshold: 1
+  min_inliers_per_pair: 10
+  min_inlier_ratio_per_pair: 0.25
+
+extractor:
+  name: "keynetaffnethardnet"
+  n_features: 8000
+  upright: False
+
+matcher:
+  name: "kornia_matcher"
+  match_mode: "smnn"
+  th: 0.95
diff --git a/config/superpoint+lightglue.yaml b/config/superpoint+lightglue.yaml
@@ -3,6 +3,7 @@
 general:
   tile_size: (2400, 2000)
   geom_verification: pydegensac
+  gv_threshold: 1
   min_inliers_per_pair: 10
   min_inlier_ratio_per_pair: 0.25
 

diff --git a/config/superpoint+superglue.yaml b/config/superpoint+superglue.yaml
@@ -3,6 +3,7 @@
 general:
   tile_size: (2400, 2000)
   geom_verification: pydegensac
+  gv_threshold: 1
   min_inliers_per_pair: 10
   min_inlier_ratio_per_pair: 0.25
 

diff --git a/src/deep_image_matching/config.py b/src/deep_image_matching/config.py
@@ -113,7 +113,7 @@
     "disk+lightglue": {
         "extractor": {
             "name": "disk",
-            "max_keypoints": 4096,
+            "max_keypoints": 8000, # CHANGED FROM 4,000 FOR LUCA EXPERIMENTS 3
         },
         "matcher": {
             "name": "lightglue",
@@ -123,7 +123,7 @@
         "extractor": {
             "name": "aliked",
             "model_name": "aliked-n16rot",
-            "max_num_keypoints": 4000,
+            "max_num_keypoints": 8000, # CHANGED FROM 4,000 FOR LUCA EXPERIMENTS 3
             "detection_threshold": 0.2,
             "nms_radius": 3,
         },