mirrors/ok-wuthering-waves
0
0
Fork 0
mirror of https://github.com/ok-oldking/ok-wuthering-waves.git synced 2025-06-03 09:25:19 +00:00
Code

优化有时会提前跳出战斗的问题

Browse Source
This commit is contained in:
firedcto@gmail.com 2024-09-24 17:50:01 +08:00
parent 4bcb995306
commit 0202c7a640
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/task/IllusiveRealmTask.py
View File

@ -4,7 +4,6 @@ import cv2
import numpy as np
from qfluentwidgets import FluentIcon
from ok.feature.FeatureSet import mask_white
from ok.logging.Logger import get_logger
from src.task.BaseCombatTask import BaseCombatTask
@ -93,3 +92,18 @@ def mask_target(image_path):
masked_image = cv2.bitwise_and(image, image, mask=combined_mask)
return masked_image
def mask_white(image):
# Check if the image is grayscale
if len(image.shape) == 2 or image.shape[2] == 1:
# Image is grayscale
lower_white = np.array([230])
upper_white = np.array([255])
else:
# Image is in color
lower_white = np.array([230, 230, 230])
upper_white = np.array([255, 255, 255])
# Create a mask for the white color
return cv2.inRange(image, lower_white, upper_white)