YoursFunny/ArisuAutoSweeper
1
0
Fork 0
mirror of https://github.com/TheFunny/ArisuAutoSweeper synced 2025-12-17 00:15:13 +00:00
Code Releases Wiki Activity

fix(cafe): crop template from button image

Browse Source
This commit is contained in:
YoursFunny 2023-11-12 15:28:54 +08:00
parent f21a97e08d
commit e5a914e80f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tasks/cafe/cafe.py
View File

@ -5,7 +5,7 @@ from enum import Enum
from module.logger import logger from module.logger import logger
from module.base.timer import Timer from module.base.timer import Timer
from module.base.button import ClickButton from module.base.button import ClickButton
from module.base.utils.utils import area_offset from module.base.utils.utils import area_offset, crop
from module.ocr.ocr import Digit from module.ocr.ocr import Digit
from module.ui.switch import Switch from module.ui.switch import Switch
from tasks.base.page import page_cafe from tasks.base.page import page_cafe
@ -63,7 +63,7 @@ class Cafe(UI):
def _match_clickable_points(self, image, threshold=0.8): def _match_clickable_points(self, image, threshold=0.8):
image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY) image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
template = cv2.cvtColor(self.image_crop(self.btn), cv2.COLOR_BGR2GRAY) template = cv2.cvtColor(crop(self.btn.matched_button.image, self.btn.area), cv2.COLOR_BGR2GRAY)
res = cv2.matchTemplate(image, template, cv2.TM_CCOEFF_NORMED) res = cv2.matchTemplate(image, template, cv2.TM_CCOEFF_NORMED)
loc = np.where(res >= threshold) loc = np.where(res >= threshold)