Crop

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

在指定位置和輸出大小裁切輸入影像。

參數:

• w (int) – 結果寬度

• h (int, optional) – 結果高度。如果為 None，則使用 w (正方形裁切)。

• top (int, optional) – 開始裁切的頂部像素座標。預設值為 0，即影像的頂部。

• left (int, optional) – 開始裁切的左側像素座標。預設值為 0，即影像的左側。

• in_keys (sequence of NestedKey, optional) – 要裁切的項目。如果未提供任何項目，則假設為 ["pixels"]。

• out_keys (sequence of NestedKey, optional) – 裁切影像的鍵。如果未提供任何鍵，則假設為 in_keys。

transform_observation_spec(observation_spec: TensorSpec) → TensorSpec

轉換 observation spec，使產生的 spec 符合轉換映射。

參數:

observation_spec (TensorSpec) – 轉換前的 spec

回傳值:

轉換後預期的 spec