CenterCrop¶
- class torchrl.envs.transforms.CenterCrop(w: int, h: Optional[int] = None, in_keys: Optional[Sequence[NestedKey]] = None, out_keys: Optional[Sequence[NestedKey]] = None)[來源]¶
裁剪影像的中心。
- 參數:
w (int) – 產生的寬度
h (int, optional) – 產生的高度。如果為 None，則使用 w (正方形裁剪)。
in_keys (NestedKey 的序列, optional) – 要裁剪的條目。如果未提供，則假定為
["pixels"]。out_keys (NestedKey 的序列, optional) – 裁剪的影像鍵。如果未提供，則假定為
in_keys。
- transform_observation_spec(observation_spec: TensorSpec) TensorSpec[來源]¶
轉換觀察規格，使產生的規格符合轉換對應。
- 參數:
observation_spec (TensorSpec) – 轉換前的規格
- 傳回:
轉換後預期的規格
