mower-ng/mower-ng
4
0
Fork 0
Code Issues 51 Pull requests Wiki Activity

Merge branch 'main' of https://git.zhaozuohong.vip/mower-ng/mower-ng

Browse source
This commit is contained in:
EightyDollars 2024-10-18 16:51:39 +08:00
commit a187435be5
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
mower/solvers/sss/sss_fight.py
View file

@ -2,6 +2,7 @@ from copy import deepcopy
from datetime import datetime, timedelta from datetime import datetime, timedelta
import cv2 import cv2
import numpy as np
from scipy.signal import argrelmax from scipy.signal import argrelmax
from skimage.metrics import structural_similarity from skimage.metrics import structural_similarity
@ -209,7 +210,11 @@ class SSSFightSolver(SceneGraphSolver):
pos = self.calc.get_character_screen_pos(x, y, False, False) pos = self.calc.get_character_screen_pos(x, y, False, False)
pos = int(pos.x), int(pos.y) pos = int(pos.x), int(pos.y)
img = cropimg(config.recog.img, sa(((-15, -168), (20, -128)), pos)) img = cropimg(config.recog.img, sa(((-15, -168), (20, -128)), pos))
result = cv2.matchTemplate(img, skill_ready, cv2.TM_CCORR_NORMED) mask = np.ones_like(skill_ready, dtype=np.float32)
mask[8:17, :] = 0
result = cv2.matchTemplate(
img, skill_ready, cv2.TM_CCOEFF_NORMED, mask=mask
)
min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(result) min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(result)
logger.debug(f"{max_val=} {max_loc=}") logger.debug(f"{max_val=} {max_loc=}")
return max_val >= 0.8 return max_val >= 0.8