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

Add: Item tag detection

Browse Source
This commit is contained in:
LmeSzinc
2021-07-20 21:27:59 +08:00
parent 70ae82e48e
commit 5d4cbe031e
2 changed files with 53 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
module/statistics/get_items.py
View File

@@ -20,7 +20,7 @@ def merge_get_items(item_list_1, item_list_2):
Returns:
list[Item]:
"""
pass
return list(set(item_list_1 + item_list_2))
class GetItemsStatistics:
@@ -51,11 +51,9 @@ class GetItemsStatistics:
elif GET_ITEMS_3.match(image, offset=(5, 0)):
ITEM_GROUP.grids = ITEM_GRIDS_3
else:
from PIL import Image
Image.fromarray(GET_ITEMS_1.image).show()
raise ImageError('Stat image is not a get_items image')
def stats_get_items(self, image):
def stats_get_items(self, image, **kwargs):
"""
Args:
image: Pillow image.
@@ -68,7 +66,7 @@ class GetItemsStatistics:
if ITEM_GROUP.grids is None:
return []
else:
ITEM_GROUP.predict(image)
ITEM_GROUP.predict(image, **kwargs)
return ITEM_GROUP.items
def load_template_folder(self, folder):