mirror of
https://github.com/W1NDes/M-AzurLaneAutoScript.git
synced 2026-05-20 17:08:20 +08:00
22 lines
550 B
Python
22 lines
550 B
Python
from module.ocr.ocr import Ocr
|
|
from module.statistics.assets import ENEMY_NAME
|
|
|
|
ENEMY_NAME_OCR = Ocr(ENEMY_NAME, lang='cnocr', threshold=128)
|
|
|
|
|
|
class BattleStatusStatistics:
|
|
def stats_battle_status(self, image):
|
|
"""
|
|
Args:
|
|
image: Pillow image.
|
|
|
|
Returns:
|
|
str: Enemy name, such as '中型主力舰队'.
|
|
"""
|
|
result = ENEMY_NAME_OCR.ocr(image)
|
|
# Delete wrong OCR result
|
|
for letter in '-一个―~(':
|
|
result = result.replace(letter, '')
|
|
|
|
return result
|