d3rlpy.augmentation.image.RandomRotation

class d3rlpy.augmentation.image.RandomRotation(degree=5.0)

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.

degree

range of degrees to rotate image.

Type:float

Methods

get_params(deep=False)

Returns augmentation parameters.

Parameters:deep (bool) – flag to deeply copy objects. Returns:augmentation parameters. Return type:dict

get_type()

Returns augmentation type.

Returns:random_rotation. Return type:str

transform(x)

Returns rotated image.

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