[update] 選択肢の認識処理修正

グレースケール画像も処理する
Cilda · Nov 3, 2022 · 367b461 · 367b461
1 parent 28049d6
commit 367b461
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/UmaUmaChecker/Uma.cpp b/UmaUmaChecker/Uma.cpp
@@ -473,7 +473,14 @@ std::shared_ptr<EventSource> Uma::GetEventByBottomOption(const cv::Mat& srcImg)
 	cv::threshold(gray, bin, 90, 255, cv::THRESH_BINARY);
 
 	std::wstring text = GetTextFromImage(bin);
-	return SkillLib.RetrieveEventFromOptionTitle(text);
+	auto event = SkillLib.RetrieveEventFromOptionTitle(text);
+	if (event) return event;
+
+	text = GetTextFromImage(gray);
+	event = SkillLib.RetrieveEventFromOptionTitle(text);
+	if (event) return event;
+
+	return nullptr;
 }
 
 std::shared_ptr<EventSource> Uma::GetCharaEventByBottomOption(const cv::Mat& srcImg)