fix: 修复特征匹配中透视转换崩溃问题

fix #275
MaaXYZ · Jul 25, 2024 · 7b1fc55 · 7b1fc55
1 parent 19fb8f2
commit 7b1fc55
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/source/MaaFramework/Vision/FeatureMatcher.cpp b/source/MaaFramework/Vision/FeatureMatcher.cpp
@@ -224,14 +224,19 @@ FeatureMatcher::ResultsVec FeatureMatcher::feature_postproc(
         return {};
     }
 
-    cv::Mat H = cv::findHomography(obj, scene, cv::RANSAC);
+    cv::Mat homography = cv::findHomography(obj, scene, cv::RHO);
+
+    if (homography.empty()) {
+        LogTrace << name_ << VAR(uid_) << "Homography is empty";
+        return {};
+    }
 
     std::array<cv::Point2d, 4> obj_corners = { cv::Point2d(0, 0),
                                                cv::Point2d(templ_cols, 0),
                                                cv::Point2d(templ_cols, templ_rows),
                                                cv::Point2d(0, templ_rows) };
     std::array<cv::Point2d, 4> scene_corners;
-    cv::perspectiveTransform(obj_corners, scene_corners, H);
+    cv::perspectiveTransform(obj_corners, scene_corners, homography);
 
     double x = std::min(
         { scene_corners[0].x, scene_corners[1].x, scene_corners[2].x, scene_corners[3].x });
@@ -336,4 +341,4 @@ void FeatureMatcher::sort_(ResultsVec& results) const
     }
 }
 
-MAA_VISION_NS_END
+MAA_VISION_NS_END