diff --git a/flower_game_env.py b/flower_game_env.py index 036ab89..e22472f 100644 --- a/flower_game_env.py +++ b/flower_game_env.py @@ -180,6 +180,12 @@ class FlowerGameEnv(gym.Env): # ---------------- Erkennung ---------------- def _grab_bgr(self): raw = np.array(self.sct.grab(self.monitor_area)) # BGRA + rgb_data = raw[:, :, :3] + target_color = np.array([84, 111, 113]) + tolerance = 20 + found_blue = np.any(np.all(np.abs(rgb_data - target_color) <= tolerance, axis=-1)) + if found_blue: + pyautogui.hotkey("ctrl", "p") return cv2.cvtColor(raw, cv2.COLOR_BGRA2BGR) def _detect_entities(self, frame_bgr): diff --git a/train_bot.py b/train_bot.py index 777f256..d36dd53 100644 --- a/train_bot.py +++ b/train_bot.py @@ -8,7 +8,7 @@ from flower_game_env import FlowerGameEnv # ---- Dein Spielbereich (anpassen!) ---- monitor_area = {"top": 120, "left": 330, "width": 1900, "height": 1263} env = FlowerGameEnv(monitor_area) -saved_model_name = "flower_bot" +saved_model_name = "flower_bottig" zip_file = saved_model_name + ".zip"