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

Opt: Template.match_multi returns list of Button

Browse Source
This commit is contained in:
LmeSzinc
2021-06-30 15:39:06 +08:00
parent be1c6420d6
commit e6c17171c8
4 changed files with 61 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
module/campaign/campaign_ocr.py
View File

@@ -85,19 +85,14 @@ class CampaignOcr(ModuleBase):
"""
digits = []
stage_image = image if stage_image is None else stage_image
result = template.match_multi(stage_image, similarity=0.85)
result = Points(result).group()
for point in result:
button = tuple(np.append(point, point + template.size))
point = point + name_offset
name = image.crop(np.append(point, point + name_size))
name = extract_letters(name, letter=name_letter, threshold=name_thresh)
stage = self._extract_stage_name(name)
area = area_offset(stage, point)
color = get_color(image, button)
digits.append(Button(area=area, color=color, button=button, name='stage'))
result = template.match_multi(stage_image, similarity=0.85, name='STAGE')
name_area = (name_offset[0], name_offset[1], name_offset[0] + name_size[0], name_offset[1] + name_size[1])
for button in result:
button_name = button.crop(area=name_area, image=image)
name = extract_letters(button_name.image, letter=name_letter, threshold=name_thresh)
button_name = button_name.crop(area=self._extract_stage_name(name))
button.area = button_name.area
digits.append(button)
return digits