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(sweep): filter non-digit ocr text

Browse Source
This commit is contained in:
YoursFunny 2023-11-21 13:14:30 +08:00
parent 30e8c8b21b
commit e61afaf43b
Signed by: YoursFunny
GPG Key ID: 207EDC3CD5B40F9C
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tasks/stage/sweep.py
View File

@ -1,8 +1,9 @@
from enum import Enum
from module.base.base import ModuleBase from module.base.base import ModuleBase
from module.base.timer import Timer from module.base.timer import Timer
from module.logger import logger from module.logger import logger
from module.ocr.ocr import Digit from module.ocr.ocr import Digit
from enum import Enum
from tasks.stage.assets.assets_stage_sweep import * from tasks.stage.assets.assets_stage_sweep import *
@ -123,7 +124,7 @@ class StageSweep:
main.device.screenshot() main.device.screenshot()
if not timer.reached_and_reset(): if not timer.reached_and_reset():
continue continue
ocr_result = self.num.detect_and_ocr(main.device.image) ocr_result = list(filter(lambda x: x.ocr_text.isdigit(), self.num.detect_and_ocr(main.device.image)))
if not ocr_result: if not ocr_result:
logger.warning(f'No valid num in {self.num.name}') logger.warning(f'No valid num in {self.num.name}')
continue continue