From 0202c7a640806836849d7b6023308142dfb1ca35 Mon Sep 17 00:00:00 2001 From: "firedcto@gmail.com" Date: Tue, 24 Sep 2024 17:50:01 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E6=9C=89=E6=97=B6=E4=BC=9A?= =?UTF-8?q?=E6=8F=90=E5=89=8D=E8=B7=B3=E5=87=BA=E6=88=98=E6=96=97=E7=9A=84?= =?UTF-8?q?=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/task/IllusiveRealmTask.py | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/src/task/IllusiveRealmTask.py b/src/task/IllusiveRealmTask.py index f6b71a8..fb3937a 100644 --- a/src/task/IllusiveRealmTask.py +++ b/src/task/IllusiveRealmTask.py @@ -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)