調整大小

class torchrl.envs.transforms.Resize(w: int, h: Optional[int] = None, interpolation: str = 'bilinear', in_keys: Optional[Sequence[NestedKey]] = None, out_keys: Optional[Sequence[NestedKey]] = None)

調整像素觀察的大小。

參數:

• w (int) – 產生的寬度。

• h (int, optional) – 產生的高度。如果未提供，則採用 w 的值。

• interpolation (str) – 插值方法

範例

>>> from torchrl.envs import GymEnv
>>> t = Resize(64, 84)
>>> base_env = GymEnv("HalfCheetah-v4", from_pixels=True)
>>> env = TransformedEnv(base_env, Compose(ToTensorImage(), t))

transform_observation_spec(observation_spec: TensorSpec) → TensorSpec

轉換觀察規格，使產生的規格與轉換映射相符。

參數:

observation_spec (TensorSpec) – 轉換前的規格

傳回:

轉換後的預期規格