From db08bc92404131f44c4e4d6e14a8de07e978d103 Mon Sep 17 00:00:00 2001 From: Ashutosh Singh <40604544+ashutoshsingh0223@users.noreply.github.com> Date: Wed, 2 Aug 2023 16:30:48 +0200 Subject: [PATCH] Update supervision.py typo in if condition for scale1 check --- src/loftr/utils/supervision.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/loftr/utils/supervision.py b/src/loftr/utils/supervision.py index 8ce6e79e..4749e24a 100644 --- a/src/loftr/utils/supervision.py +++ b/src/loftr/utils/supervision.py @@ -41,7 +41,7 @@ def spvs_coarse(data, config): _, _, H1, W1 = data['image1'].shape scale = config['LOFTR']['RESOLUTION'][0] scale0 = scale * data['scale0'][:, None] if 'scale0' in data else scale - scale1 = scale * data['scale1'][:, None] if 'scale0' in data else scale + scale1 = scale * data['scale1'][:, None] if 'scale1' in data else scale h0, w0, h1, w1 = map(lambda x: x // scale, [H0, W0, H1, W1]) # 2. warp grids