d3rlpy.augmentation.image.RandomRotation¶
-
class
d3rlpy.augmentation.image.
RandomRotation
(degree=5.0)[source]¶ Random rotation augmentation.
References
Parameters: degree (float) – range of degrees to rotate image. Methods
-
get_params
(deep=False)[source]¶ Returns augmentation parameters.
Parameters: deep (bool) – flag to deeply copy objects. Returns: augmentation parameters. Return type: dict
-
transform
(x)[source]¶ Returns rotated image.
Parameters: x (torch.Tensor) – observation tensor. Returns: processed observation tensor. Return type: torch.Tensor
Attributes
-
TYPE
= 'random_rotation'¶
-