d3rlpy.augmentation.image.HorizontalFlip

class d3rlpy.augmentation.image.HorizontalFlip(probability=0.1)

Horizontal flip augmentation.

References

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

Parameters

probability (float) – probability to flip horizontally.

Methods

get_params(deep=False)

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)

Returns augmented observation.

Parameters

x (torch.Tensor) – observation.

Returns

augmented observation.

Return type

torch.Tensor

Attributes

TYPE: ClassVar[str] = 'horizontal_flip'