d3rlpy.augmentation.image.RandomRotation¶

class d3rlpy.augmentation.image.RandomRotation(degree=5.0)[source]¶

Random rotation augmentation.

References

Kostrikov et al., Image Augmentation Is All You Need: Regularizing Deep Reinforcement Learning from Pixels.

Parameters: degree (float) – range of degrees to rotate image.

Methods

get_params(deep=False)[source]¶

Returns augmentation parameters.

Parameters: deep (bool) – flag to copy parameters.
Returns: augmentation parameters.
Return type: Dict[str, Any]

get_type()¶

Returns augmentation type.

Returns: augmentation type.
Return type: str

transform(x)[source]¶

Returns augmented observation.

Parameters: x (torch.Tensor) – observation.
Returns: augmented observation.
Return type: torch.Tensor

Attributes

TYPE: ClassVar[str] = 'random_rotation'¶