Lars/turtleBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

changed to multi binary action

Browse Source
This commit is contained in:
lars 2025-08-14 16:51:11 +02:00
parent 3e266b1c35
commit 1792c1f971
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
flower_game_env.py
View File

@ -141,7 +141,7 @@ class FlowerGameEnv(gym.Env):
"state": spaces.Box(low=0.0, high=1.0, shape=(6,), dtype=np.float32) "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 --- # --- HSV-Grenzen ---
self.yellow_lower = np.array([15, 40, 200], dtype=np.uint8) 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("a", _pause=False)
pyautogui.keyUp("s", _pause=False) pyautogui.keyUp("s", _pause=False)
pyautogui.keyUp("d", _pause=False) pyautogui.keyUp("d", _pause=False)
if action == 0: if action[0] and not action[2]:
pyautogui.keyDown("w", _pause=False) pyautogui.keyDown("w", _pause=False)
elif action == 1: if action[1] and not action[3]:
pyautogui.keyDown("a", _pause=False) pyautogui.keyDown("a", _pause=False)
elif action == 2: if action[2] and not action[0]:
pyautogui.keyDown("s", _pause=False) pyautogui.keyDown("s", _pause=False)
elif action == 3: if action[3] and not action[1]:
pyautogui.keyDown("d", _pause=False) pyautogui.keyDown("d", _pause=False)
time.sleep(0.2) time.sleep(0.15)
obs = self._build_observation() obs = self._build_observation()
reward = self._calculate_reward() reward = self._calculate_reward()