Merge pull request zju3dv#286 from ashutoshsingh0223/master

Update supervision.py
avichai-airis · Aug 3, 2023 · df7ca80 · df7ca80
2 parents c0546af + db08bc9
commit df7ca80
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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