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
- Parameters
Methods
-
transform
(x)[source]¶ Returns augmented observation.
- Parameters
x (torch.Tensor) – observation.
- Returns
augmented observation.
- Return type
torch.Tensor
Attributes