视频录制器

torchrl.record.VideoRecorder(logger: Logger, tag: str, in_keys: Optional[Sequence[NestedKey]] = None, skip: int | None = None, center_crop: Optional[int] = None, make_grid: bool | None = None, out_keys: Optional[Sequence[NestedKey]] = None, **kwargs) → None

视频录制器转换。

将记录来自环境的一系列观察结果，并在需要时将其写入 Logger 对象。

参数：:

• logger (Logger) – 要写入视频的 Logger 实例。若要在 memmap 张量或 mp4 文件下保存视频，请使用 CSVLogger 类。

• tag (str) – 日志记录器中的视频标签。

• in_keys (NestedKey 的序列, 可选) – 用于生成视频的键。默认值为 "pixels"。

• skip (int) – 输出视频中的帧间隔。如果转换具有父环境，则默认值为 2，否则为 1。

• center_crop (int, 可选) – 方形中心裁剪的值。

• make_grid (bool, 可选) – 如果为 True，则会创建一个网格，假设提供了一个形状为 [B x W x H x 3] 的张量，其中 B 为批次大小。如果转换具有父环境，则默认值为 True，否则为 False。

• out_keys (NestedKey 的序列, 可选) – 目标键。如果未提供，则默认为 in_keys。

示例

以下示例演示如何将展开保存到视频中。首先，进行一些导入

>>> from torchrl.record import VideoRecorder
>>> from torchrl.record.loggers.csv import CSVLogger
>>> from torchrl.envs import TransformedEnv, DMControlEnv

视频格式是在日志记录器中选择的。Wandb 和 tensorboard 会自行处理，CSV 接受各种视频格式。

>>> logger = CSVLogger(exp_name="cheetah", log_dir="cheetah_videos", video_format="mp4")

一些环境（例如 Atari 游戏）本机返回图像，一些需要用户请求图像。检查 GymEnv 或 DMControlEnv 了解如何在这些上下文中渲染图像。

>>> base_env = DMControlEnv("cheetah", "run", from_pixels=True)
>>> env = TransformedEnv(base_env, VideoRecorder(logger=logger, tag="run_video"))
>>> env.rollout(100)

所有转换都有一个转储函数，大多数情况下是 no-op，除了 VideoRecorder 和 Compose，它们会将 dumps 分派给所有成员。

>>> env.transform.dump()

该转换也可以在数据集内使用以保存收集的视频。与环境情况不同，图像将以批次形式出现。skip 参数将能够仅在特定间隔保存图像。

>>> from torchrl.data.datasets import OpenXExperienceReplay
>>> from torchrl.envs import Compose
>>> from torchrl.record import VideoRecorder, CSVLogger
>>> # Create a logger that saves videos as mp4
>>> logger = CSVLogger("./dump", video_format="mp4")
>>> # We use the VideoRecorder transform to save register the images coming from the batch.
>>> t = VideoRecorder(logger=logger, tag="pixels", in_keys=[("next", "observation", "image")])
>>> # Each batch of data will have 10 consecutive videos of 200 frames each (maximum, since strict_length=False)
>>> dataset = OpenXExperienceReplay("cmu_stretch", batch_size=2000, slice_len=200,
...             download=True, strict_length=False,
...             transform=t)
>>> # Get a batch of data and visualize it
>>> for data in dataset:
...     t.dump()
...     break

我们的视频位于 ./cheetah_videos/cheetah/videos/run_video_0.mp4 下！