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

Fix: Handle missing / in dorm food ocr

Browse Source
This commit is contained in:
LmeSzinc
2024-02-06 20:38:54 +08:00
parent 1750f31199
commit ae855a8a74
1 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
module/dorm/dorm.py
View File

@@ -30,6 +30,21 @@ class OcrDormFood(DigitCounter):
image = cv2.multiply(image, 2) image = cv2.multiply(image, 2)
return image return image
def after_process(self, result):
result = super().after_process(result)
if '/' not in result:
for exp in range(40000, 90001, 1000):
res = re.match(rf'^(\d+){exp}$', result)
if res:
# 10005800 -> 1000/5800
new = f'{res.group(1)}/{exp}'
logger.info(f'OcrDormFood result {result} is revised to {new}')
result = new
break
return result
OCR_FILL = OcrDormFood(OCR_DORM_FILL, name='OCR_DORM_FILL') OCR_FILL = OcrDormFood(OCR_DORM_FILL, name='OCR_DORM_FILL')