tdpeuter/2026SEL3-project-Brittle_Star_Controller_Modularity
tdpeuter/2026SEL3-project-Brittle_Star_Controller_Modularity
Archived
1
Fork 0
Code Issues Releases Activity

feat: added video configs fields

Browse source
This commit is contained in:
Jona Reynaert 2026-05-14 13:39:47 +02:00
parent dd0e5b2d09
commit b84e8800ad
3 changed files with 10 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
configs/simulation/default.yaml
View file

@ -21,6 +21,10 @@ video_output_path: null
# Camera ID to use for video recording (1 is usually the close-up camera) # Camera ID to use for video recording (1 is usually the close-up camera)
camera_id: 1 camera_id: 1
video_width: int = 640
video_height: int = 480
video_fps: int = 60
# Optional override for the metadata YAML file path. # Optional override for the metadata YAML file path.
# If null, the script looks for `<model_name>_metadata.yaml` alongside the model_path. # If null, the script looks for `<model_name>_metadata.yaml` alongside the model_path.
metadata_path: null metadata_path: null

3
scripts/simulate.py
View file

@ -107,6 +107,9 @@ def main(dict_cfg: DictConfig) -> None:
action_mask=action_mask, action_mask=action_mask,
output_path=output_path, output_path=output_path,
camera_id=sim_cfg.camera_id, camera_id=sim_cfg.camera_id,
width=sim_cfg.video_width,
height=sim_cfg.video_height,
fps=sim_cfg.video_fps,
) )
save_evaluation_metadata( save_evaluation_metadata(

3
src/brittle_star_project/configs/config_simulation.py
View file

@ -26,6 +26,9 @@ class SimulationSettings:
video_output_path: Optional[str] = None video_output_path: Optional[str] = None
# Camera ID to use for video recording (1 is usually the close-up camera) # Camera ID to use for video recording (1 is usually the close-up camera)
camera_id: int = 1 camera_id: int = 1
video_width: int = 640
video_height: int = 480
video_fps: int = 60
# Optional override for the sidecar metadata YAML file. # Optional override for the sidecar metadata YAML file.
# If None, it defaults to the model_path with a `_metadata.yaml` suffix. # If None, it defaults to the model_path with a `_metadata.yaml` suffix.