sui_feng/AzurLaneAutoScript
1
0
Fork 0
mirror of https://gitee.com/sui-feng-cb/AzurLaneAutoScript1 synced 2026-03-09 18:39:04 +08:00
Code

Fix: False fleet mechanism detection on island and belt

Browse Source
This commit is contained in:
LmeSzinc 2022-05-17 23:36:48 +08:00
parent ed7c7a6fb8
commit 00587a3423
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
module/map_detection/os_grid.py
View File

@ -294,6 +294,18 @@ class OSGridPredictor(GridPredictor):
count = image[image > 221].shape[0]
if count < 30:
return False
# Shouldn't contain any thing green or yellow
# Green is island and yellow is belt
image = cv2.cvtColor(crop(self.image, area), cv2.COLOR_RGB2HSV)
h = (0, 180)
s = (30, 90)
v = (30, 100)
lower = (h[0] / 2, s[0] * 2.55, v[0] * 2.55)
upper = (h[1] / 2 + 1, s[1] * 2.55 + 1, v[1] * 2.55 + 1)
image_in_range = cv2.inRange(image, lower, upper)
if image_in_range[image_in_range > 0].shape[0] > 30:
return False
return True