minor changes

3DOM-FBK · Sep 26, 2024 · 2d91358 · 2d91358
1 parent d75963b
commit 2d91358
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/deep_image_matching/matchers/srif.py b/src/deep_image_matching/matchers/srif.py
@@ -95,7 +95,7 @@ def _match_pairs(
         Raises:
             torch.cuda.OutOfMemoryError: If an out-of-memory error occurs while matching images.
         """
-        MAX_FEATURES = 5000
+        MAX_FEATURES = 8000
         img0_name = img0_path.name
         img1_name = img1_path.name
 
@@ -126,7 +126,7 @@ def _match_pairs(
                 mkpts1,
                 px_th=1.0,
                 conf=0.99,
-                max_iters=100000,
+                max_iters=10000,
             )
             mkpts0 = mkpts0[inlMask, :]
             mkpts1 = mkpts1[inlMask, :]