From df402f9c4cd88fb03c9374f04cf27ba1966bdabf Mon Sep 17 00:00:00 2001
From: sp <stefan.pranger@iaik.tugraz.at>
Date: Tue, 11 Feb 2025 14:14:25 +0100
Subject: [PATCH] removed VideoCallback
and moviepy dependency
---
dockerfile | 1 -
notebooks/sb3utils.py | 46 -------------------------------------------
2 files changed, 47 deletions(-)
diff --git a/dockerfile b/dockerfile
index c3a5e55..38a3444 100644
--- a/dockerfile
+++ b/dockerfile
@@ -83,7 +83,6 @@ RUN pip install ipywidgets
RUN pip install matplotlib
RUN pip install sb3-contrib
RUN pip install opencv-python
-RUN pip install moviepy
RUN pip install gymnasium==0.29.0
RUN pip install numpy==1.24.4
diff --git a/notebooks/sb3utils.py b/notebooks/sb3utils.py
index 9ed75b8..97bae3a 100644
--- a/notebooks/sb3utils.py
+++ b/notebooks/sb3utils.py
@@ -1,7 +1,6 @@
import gymnasium as gym
import numpy as np
import random
-from moviepy.editor import ImageSequenceClip
from utils import MiniGridShieldHandler, common_parser
from stable_baselines3.common.callbacks import BaseCallback, CheckpointCallback
@@ -45,51 +44,6 @@ def parse_sb3_arguments():
return args
-class ImageRecorderCallback(BaseCallback):
- def __init__(self, eval_env, render_freq, n_eval_episodes, evaluation_method, log_dir, deterministic=True, verbose=0):
- super().__init__(verbose)
-
- self._eval_env = eval_env
- self._render_freq = render_freq
- self._n_eval_episodes = n_eval_episodes
- self._deterministic = deterministic
- self._evaluation_method = evaluation_method
- self._log_dir = log_dir
-
- def _on_training_start(self):
- image = self.training_env.render(mode="rgb_array")
- self.logger.record("trajectory/image", Image(image, "HWC"), exclude=("stdout", "log", "json", "csv"))
-
- def _on_step(self) -> bool:
- #if self.n_calls % self._render_freq == 0:
- # self.record_video()
- return True
-
- def _on_training_end(self) -> None:
- self.record_video()
-
- def record_video(self) -> bool:
- screens = []
- def grab_screens(_locals, _globals) -> None:
- """
- Renders the environment in its current state, recording the screen in the captured `screens` list
-
- :param _locals: A dictionary containing all local variables of the callback's scope
- :param _globals: A dictionary containing all global variables of the callback's scope
- """
- screen = self._eval_env.render()
- screens.append(screen)
- self._evaluation_method(
- self.model,
- self._eval_env,
- callback=grab_screens,
- n_eval_episodes=self._n_eval_episodes,
- deterministic=self._deterministic,
- )
-
- clip = ImageSequenceClip(list(screens), fps=3)
- clip.write_gif(f"{self._log_dir}/{self.n_calls}.gif", fps=3)
- return True
class InfoCallback(BaseCallback):