Resize

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

調整畫素觀測的大小。

引數:

• w (int) – 結果寬度。

• h (int, 可選) – 結果高度。如果未提供，則使用 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

轉換觀測 Spec，使結果 Spec 與 Transform 對映相匹配。

引數:

observation_spec (TensorSpec) – Transform 前的 Spec

返回值:

Transform 後的預期 Spec