Skip to content

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