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.

shift_size

size to shift image.

Type:int

Methods

get_params()

Returns augmentation parameters.

Returns:augmentation parameters. Return type:dict

get_type()

Returns augmentation type.

Returns:random_shift. Return type:str

transform(x)

Returns shifted images.

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