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) – 轉換前的規範

返回：

轉換後的預期規範