From 2d91358ae78d3d33453c3fc13beb03feda369b7d Mon Sep 17 00:00:00 2001 From: lcmrl Date: Thu, 26 Sep 2024 16:50:42 +0200 Subject: [PATCH] minor changes --- src/deep_image_matching/matchers/srif.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/deep_image_matching/matchers/srif.py b/src/deep_image_matching/matchers/srif.py index ecbd44a..553830a 100644 --- 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, :]