From 9c68ab01e68e6b0471daec0a134db9a343b070ca Mon Sep 17 00:00:00 2001 From: YoursFunny Date: Sat, 11 Nov 2023 23:45:08 +0800 Subject: [PATCH] fix(cafe): apply mask --- tasks/cafe/cafe.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tasks/cafe/cafe.py b/tasks/cafe/cafe.py index 525069a..e02d9df 100644 --- a/tasks/cafe/cafe.py +++ b/tasks/cafe/cafe.py @@ -72,7 +72,7 @@ class Cafe(UI): def _get_clickable_buttons(self, threshold=0.8, offset=(0, 0)): image = self.device.image h, w = image.shape[:2] - cv2.rectangle(image, (0, 10), (w - 25, h - 10), (0, 0, 0), 50) + image = cv2.rectangle(image, (0, 10), (w - 25, h - 10), (0, 0, 0), 50) image = self._extract_clickable_from_image(image) points = self._match_clickable_points(image, threshold) points = self.merge_points(points)