mirror of
https://gitee.com/sui-feng-cb/AzurLaneAutoScript1
synced 2026-03-09 18:39:04 +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
|