d3rlpy.augmentation.image.ColorJitter¶

class d3rlpy.augmentation.image.ColorJitter(brightness=(0.6, 1.4), contrast=(0.6, 1.4), saturation=(0.6, 1.4), hue=(-0.5, 0.5))[source]¶

Color Jitter augmentation.

This augmentation modifies the given images in the HSV channel spaces as well as a contrast change. This augmentation will be useful with the real world images.

References

Laskin et al., Reinforcement Learning with Augmented Data.

Parameters:	brightness (tuple) – brightness scale range. contrast (tuple) – contrast scale range. saturation (tuple) – saturation scale range. hue (tuple) – hue scale range.

brightness¶

brightness scale range.

Type:	tuple

contrast¶

contrast scale range.

Type:	tuple

saturation¶

saturation scale range.

Type:	tuple

hue¶

hue scale range.

Type:	tuple

Methods

get_params()[source]¶

Returns augmentation parameters.

Returns:	augmentation parameters.
Return type:	dict

get_type()[source]¶

Returns augmentation type.

Returns:	color_jitter.
Return type:	str

transform(x)[source]¶

Returns jittered images.

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