d3rlpy.augmentation.image.HorizontalFlip¶
-
class
d3rlpy.augmentation.image.
HorizontalFlip
(probability=0.1)[source]¶ Horizontal flip augmentation.
References
Parameters: probability (float) – probability to flip horizontally. 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 horizontally flipped image.
Parameters: x (torch.Tensor) – observation tensor. Returns: processed observation tensor. Return type: torch.Tensor
Attributes
-
TYPE
= 'horizontal_flip'¶
-