added RANSAC in geometric_verification

Former-commit-id: ecec9a2

config/roma.yaml

@@ -0,0 +1,13 @@
+# User configuration file
+
+general:
+  tile_size: (2400, 2000)
+  geom_verification: ransac
+  min_inliers_per_pair: 10
+  min_inlier_ratio_per_pair: 0.25
+
+extractor:
+  name: "no_extractor"
+
+matcher:
+  name: "roma"

src/deep_image_matching/__init__.py

@@ -25,6 +25,7 @@ class GeometricVerification(Enum):
     NONE = 0
     PYDEGENSAC = 1
     MAGSAC = 2
+    RANSAC = 3
 
 
 class Quality(Enum):

src/deep_image_matching/utils/geometric_verification.py

@@ -103,4 +103,20 @@ def geometric_verification(
             )
             inlMask = np.ones(len(kpts0), dtype=bool)
 
+    if method == GeometricVerification.RANSAC:
+        try:
+            F, inliers = cv2.findFundamentalMat(
+                kpts0, kpts1, cv2.RANSAC, threshold, confidence, max_iters
+            )
+            inlMask = (inliers > 0).squeeze()
+            if not quiet:
+                logger.debug(
+                    f"RANSAC found {inlMask.sum()} inliers ({inlMask.sum()*100/len(kpts0):.2f}%)"
+                )
+        except Exception as err:
+            logger.error(
+                f"{err}. Unable to perform geometric verification with RANSAC."
+            )
+            inlMask = np.ones(len(kpts0), dtype=bool)
+
     return F, inlMask