RandomColorDistortion
RandomColorDistortion class
dlf.preprocessing.color_distortion.RandomColorDistortion(
s=1.0, color_jitter_probability=0.8, color_drop_probability=0.2
)
Data augmentation method that randomly distorts colors of an RGB image
Aliases
- color_distortion
- ColorDistortion
- random_color_distortion
- RandomColorDistortion
Arguments
- s: float. Scalar that scales the distortion effect. Defaults to 1.0.
- color_jitter_probability: float [0,1]. Probability that color jittering is applied. Defaults to 0.8
- color_drop_probability: float [0,1]. Probability that color drop (RGB to Gray) is applied. Defaults to 0.8
YAML Configuration
preprocess_list:
color_distortion:
s: 0.7
color_jitter_probability: 0.5
color_drop_probability: 0.3
References
- SimCLR: https://arxiv.org/abs/2002.05709