sui_feng/AzurLaneAutoScript
1
0
Fork 0
mirror of https://gitee.com/sui-feng-cb/AzurLaneAutoScript1 synced 2026-03-15 19:17:21 +08:00
Code Activity

upd: [JP] LevelOcr and EmotionDigit

Browse Source
This commit is contained in:
guoh064
2024-10-19 11:32:31 +08:00
committed by LmeSzinc
parent d1052df932
commit 65304fee2e
3 changed files with 43 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
module/retire/scanner.py
View File

@@ -5,8 +5,9 @@ from typing import Any, Dict, List, Tuple, Union
import cv2
import numpy as np
import module.config.server as server
from module.base.button import ButtonGrid
from module.base.utils import color_similar, crop, get_color, limit_in
from module.base.utils import color_similar, crop, extract_letters, get_color, limit_in, save_image
from module.combat.level import LevelOcr
from module.logger import logger
from module.ocr.ocr import Digit
@@ -20,6 +21,15 @@ from module.retire.dock import (CARD_EMOTION_GRIDS, CARD_GRIDS,
class EmotionDigit(Digit):
def pre_process(self, image):
if server.server == 'jp':
image_gray = extract_letters(image, letter=(255, 255, 255), threshold=self.threshold)
right_side = np.nonzero(image_gray[0:16, :].min(axis=0) > 176)[-1]
image = image[:, :right_side[-1]]
image = super().pre_process(image)
return image
def after_process(self, result):
# Random OCR error on Downes' hair
# OCR DOCK_EMOTION_OCR: Result "044" is revised to "44"
@@ -132,8 +142,14 @@ class EmotionScanner(Scanner):
super().__init__()
self._results = []
self.grids = CARD_EMOTION_GRIDS
self.ocr_model = EmotionDigit(self.grids.buttons,
if server.server != 'jp':
self.ocr_model = EmotionDigit(self.grids.buttons,
name='DOCK_EMOTION_OCR', threshold=176)
else:
self.ocr_model = EmotionDigit(self.grids.buttons,
name='DOCK_EMOTION_OCR',
letter=(201, 201, 201),
threshold=176)
def _scan(self, image) -> List:
return self.ocr_model.ocr(image)