1
0
mirror of https://github.com/TheFunny/ArisuAutoSweeper synced 2026-06-09 17:54:52 +00:00

fix(sweep): filter non-digit ocr text

This commit is contained in:
2023-11-21 13:14:30 +08:00
parent 30e8c8b21b
commit e61afaf43b
+3 -2
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