WindyMadman/AzurLaneAutoScript
1
0
Fork 0
mirror of https://github.com/LmeSzinc/AzurLaneAutoScript.git synced 2025-01-09 05:47:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: Handle random OCR error on Downes' hair

Browse Source
This commit is contained in:
LmeSzinc 2023-05-13 21:34:17 +08:00
parent 5a717e44ad
commit f77f273329
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
module/retire/scanner.py
View File

@ -19,6 +19,16 @@ from module.retire.dock import (CARD_EMOTION_GRIDS, CARD_GRIDS,
CARD_LEVEL_GRIDS, CARD_RARITY_GRIDS)
class EmotionDigit(Digit):
def after_process(self, result):
# Random OCR error on Downes' hair
# OCR DOCK_EMOTION_OCR: Result "044" is revised to "44"
if result == '044' or result == 'D44':
result = '0'
return super().after_process(result)
@dataclass(frozen=True)
class Ship:
rarity: str = ''
@ -122,8 +132,8 @@ class EmotionScanner(Scanner):
super().__init__()
self._results = []
self.grids = CARD_EMOTION_GRIDS
self.ocr_model = Digit(self.grids.buttons,
name='DOCK_EMOTION_OCR', threshold=176)
self.ocr_model = EmotionDigit(self.grids.buttons,
name='DOCK_EMOTION_OCR', threshold=176)
def _scan(self, image) -> List:
return self.ocr_model.ocr(image)