CenterCrop

class torchrl.envs.transforms.CenterCrop(w: int, h: int = None, in_keys: Sequence[NestedKey] | None = None, out_keys: Sequence[NestedKey] | None = None)

裁剪图像中心部分。

参数：

• w (int) – 结果宽度

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

• in_keys (sequence of NestedKey, 可选) – 要裁剪的条目。如果未提供，则默认为 ["pixels"]。

• out_keys (sequence of NestedKey, 可选) – 裁剪后图像的键。如果未提供，则默认为 in_keys。

transform_observation_spec(observation_spec: TensorSpec) → TensorSpec

转换观察空间规范 (observation spec)，使结果规范与转换映射匹配。

参数：

observation_spec (TensorSpec) – 转换前的规范

返回：

转换后的预期规范