fz-mirrors/MaaAssistantArknights
0
0
Fork 0
mirror of https://github.com/MaaAssistantArknights/MaaAssistantArknights.git synced 2026-07-15 17:30:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: GenerateMaskRange 可以直接展示 mask_range 掩码后的图片

Browse Source
This commit is contained in:
zzyyyl
2024-07-26 01:32:13 +08:00
parent c69e212be9
commit 02a397c749
1 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tools/GenerateMaskRange/generate_mask_range.py
View File

@@ -77,11 +77,24 @@ def show_image_mask(image, mask, image_for_hist):
plt.tight_layout()
plt.show()
def calc_mask_from_ranges(image, mask_ranges):
mask = None
for mask_range in mask_ranges:
l0, l1, l2 = mask_range[0]
u0, u1, u2 = mask_range[1]
if mask is not None:
mask = cv2.bitwise_or(mask, cv2.inRange(image, (l0, l1, l2), (u0, u1, u2)))
else:
mask = cv2.inRange(image, (l0, l1, l2), (u0, u1, u2))
return mask
if __name__ == '__main__':
image = cv2.imread(maa_dir / "resource" / "template" / "Sarkaz@Roguelike@StageCombatDpsEnter.png")
image = cv2.imread(maa_dir / "resource" / "template" / "Sarkaz@Roguelike@StageFilterTruthEnter.png")
hsv_image = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
rgb_image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
show_image_mask(image, calc_mask_from_ranges(hsv_image, [[[23, 150, 40], [23, 230, 150]]]), hsv_image)
# 自定义参数
hsv_base_mask = None
# hsv_base_mask = cv2.inRange(hsv_image, (0, 0, 0), (180, 254, 230)) # 忽略亮色背景