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

added automatic restart

Browse Source
This commit is contained in:
lars 2025-08-10 18:51:54 +02:00
parent 2d43078a19
commit f1d1b14f2a
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
flower_game_env.py
View File

@ -180,6 +180,12 @@ class FlowerGameEnv(gym.Env):
# ---------------- Erkennung ---------------- # ---------------- Erkennung ----------------
def _grab_bgr(self): def _grab_bgr(self):
raw = np.array(self.sct.grab(self.monitor_area)) # BGRA 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) return cv2.cvtColor(raw, cv2.COLOR_BGRA2BGR)
def _detect_entities(self, frame_bgr): def _detect_entities(self, frame_bgr):

2
train_bot.py
View File

@ -8,7 +8,7 @@ from flower_game_env import FlowerGameEnv
# ---- Dein Spielbereich (anpassen!) ---- # ---- Dein Spielbereich (anpassen!) ----
monitor_area = {"top": 120, "left": 330, "width": 1900, "height": 1263} monitor_area = {"top": 120, "left": 330, "width": 1900, "height": 1263}
env = FlowerGameEnv(monitor_area) env = FlowerGameEnv(monitor_area)
saved_model_name = "flower_bot" saved_model_name = "flower_bottig"
zip_file = saved_model_name + ".zip" zip_file = saved_model_name + ".zip"