feat: added video configs fields
This commit is contained in:
parent
dd0e5b2d09
commit
b84e8800ad
3 changed files with 10 additions and 0 deletions
|
|
@ -21,6 +21,10 @@ video_output_path: null
|
|||
# Camera ID to use for video recording (1 is usually the close-up camera)
|
||||
camera_id: 1
|
||||
|
||||
video_width: int = 640
|
||||
video_height: int = 480
|
||||
video_fps: int = 60
|
||||
|
||||
# Optional override for the metadata YAML file path.
|
||||
# If null, the script looks for `<model_name>_metadata.yaml` alongside the model_path.
|
||||
metadata_path: null
|
||||
|
|
|
|||
|
|
@ -107,6 +107,9 @@ def main(dict_cfg: DictConfig) -> None:
|
|||
action_mask=action_mask,
|
||||
output_path=output_path,
|
||||
camera_id=sim_cfg.camera_id,
|
||||
width=sim_cfg.video_width,
|
||||
height=sim_cfg.video_height,
|
||||
fps=sim_cfg.video_fps,
|
||||
)
|
||||
|
||||
save_evaluation_metadata(
|
||||
|
|
|
|||
|
|
@ -26,6 +26,9 @@ class SimulationSettings:
|
|||
video_output_path: Optional[str] = None
|
||||
# Camera ID to use for video recording (1 is usually the close-up camera)
|
||||
camera_id: int = 1
|
||||
video_width: int = 640
|
||||
video_height: int = 480
|
||||
video_fps: int = 60
|
||||
|
||||
# Optional override for the sidecar metadata YAML file.
|
||||
# If None, it defaults to the model_path with a `_metadata.yaml` suffix.
|
||||
|
|
|
|||
Reference in a new issue