d3rlpy.augmentation.image.RandomShift

class d3rlpy.augmentation.image.RandomShift(shift_size=4)

Random shift augmentation.

References

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

Parameters

shift_size (int) – size to shift image.

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] = 'random_shift'