裁剪

类 torchrl.envs.transforms.Crop(w: int, h: int = None, top: int = 0, left: int = 0, in_keys: Sequence[NestedKey] | None = None, out_keys: Sequence[NestedKey] | None = None)

在指定位置和输出尺寸处裁剪输入图像。

参数:

• w (int) – 结果宽度

• h (int, 可选) – 结果高度。如果为 None，则使用 w（正方形裁剪）。

• top (int, 可选) – 开始裁剪的顶部像素坐标。默认为 0，即图像顶部。

• left (int, 可选) – 开始裁剪的左侧像素坐标。默认为 0，即图像左侧。

• in_keys (NestedKey 序列, 可选) – 要裁剪的条目。如果未提供，则假定为 ["pixels"]。

• out_keys (NestedKey 序列, 可选) – 裁剪后图像的键。如果未提供，则假定为 in_keys。

transform_observation_spec(observation_spec: TensorSpec) → TensorSpec

变换观察规范，使结果规范与变换映射匹配。

参数:

observation_spec (TensorSpec) – 变换前的规范

返回:

变换后的预期规范