d3rlpy.augmentation.image.Cutout

class d3rlpy.augmentation.image.Cutout(probability=0.5)

Cutout augmentation.

References

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

Parameters:probability (float) – probability to cutout.

probability

probability to cutout.

Type:float

Methods

get_params()

Returns augmentation parameters.

Returns:augmentation parameters. Return type:dict

get_type()

Returns augmentation type.

Returns:cutout. Return type:str

transform(x)

Returns observation performed Cutout.

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