Fixed homography RANSAC filtering in matching demo

rpautrat · Jan 6, 2023 · 1742343 · 1742343
1 parent b73aa8a
commit 1742343
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/superpoint/match_features_demo.py b/superpoint/match_features_demo.py
@@ -60,8 +60,8 @@ def compute_homography(matched_kp1, matched_kp2):
     matched_pts2 = cv2.KeyPoint_convert(matched_kp2)
 
     # Estimate the homography between the matches using RANSAC
-    H, inliers = cv2.findHomography(matched_pts1[:, [1, 0]],
-                                    matched_pts2[:, [1, 0]],
+    H, inliers = cv2.findHomography(matched_pts1,
+                                    matched_pts2,
                                     cv2.RANSAC)
     inliers = inliers.flatten()
     return H, inliers