d3rlpy.augmentation.image.Intensity

class d3rlpy.augmentation.image.Intensity(scale=0.1)

Intensity augmentation.

\[x' = x + n\]

where \(n \sim N(0, scale)\).

References

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

Parameters:scale (float) – scale of multiplier.

scale

scale of multiplier.

Type:float

Methods

get_params()

Returns augmentation parameters.

Returns:augmentation parameters. Return type:dict

get_type()

Returns augmentation type.

Returns:intensity. Return type:str

transform(x)

Returns multiplied image.

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