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

Fix: Fleet Mechanism detection

Browse Source 
- Fix: Special whether on opsi map are detected as fleet mechanism
This commit is contained in:
LmeSzinc
2022-06-03 00:26:24 +08:00
parent c669ceeb87
commit 515beddde5
6 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
module/map_detection/os_grid.py
View File

@@ -296,14 +296,21 @@ class OSGridPredictor(GridPredictor):
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:
# 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
# Should match the letter `2`
image = rgb2gray(self.image_trans)
sim, button = TEMPLATE_FleetMechanism.match_result(image)
point = (53, 37)
distance = np.linalg.norm(np.subtract(button.area[:2], point))
if distance > 5 or sim < 0.3:
return False
return True