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)