GrayScale¶
- class torchrl.envs.transforms.GrayScale(in_keys: collections.abc.Sequence[tensordict._nestedkey.NestedKey] | None = None, out_keys: collections.abc.Sequence[tensordict._nestedkey.NestedKey] | None = None)[source]¶
Turns a pixel observation to grayscale.
- transform_observation_spec(observation_spec: TensorSpec) TensorSpec[source]¶
Transforms the observation spec such that the resulting spec matches transform mapping.
- Parameters:
observation_spec (TensorSpec) – spec before the transform
- Returns:
expected spec after the transform
