From f21a97e08d66a4a51e98927bc77e6d79965b414f 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)