From 1792c1f971ae28f26699f8616457894b73485e84 Mon Sep 17 00:00:00 2001 From: lars Date: Thu, 14 Aug 2025 16:51:11 +0200 Subject: [PATCH] changed to multi binary action --- flower_game_env.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/flower_game_env.py b/flower_game_env.py index a28a34d..19a5a4b 100644 --- a/flower_game_env.py +++ b/flower_game_env.py @@ -141,7 +141,7 @@ class FlowerGameEnv(gym.Env): "state": spaces.Box(low=0.0, high=1.0, shape=(6,), dtype=np.float32) } ) - self.action_space = spaces.Discrete(4) + self.action_space = spaces.MultiBinary(4) # --- HSV-Grenzen --- self.yellow_lower = np.array([15, 40, 200], dtype=np.uint8) @@ -230,16 +230,16 @@ class FlowerGameEnv(gym.Env): pyautogui.keyUp("a", _pause=False) pyautogui.keyUp("s", _pause=False) pyautogui.keyUp("d", _pause=False) - if action == 0: + if action[0] and not action[2]: pyautogui.keyDown("w", _pause=False) - elif action == 1: + if action[1] and not action[3]: pyautogui.keyDown("a", _pause=False) - elif action == 2: + if action[2] and not action[0]: pyautogui.keyDown("s", _pause=False) - elif action == 3: + if action[3] and not action[1]: pyautogui.keyDown("d", _pause=False) - time.sleep(0.2) + time.sleep(0.15) obs = self._build_observation() reward = self._calculate_reward()