ColorJitter — Torchvision main documentation - pytorch.org
pytorch.org/vision/main/generated/torchvision.transforms.ColorJitter.htmlColorJitter¶ class torchvision.transforms. ColorJitter (brightness = 0, contrast = 0, saturation = 0, hue = 0) [source] ¶. Randomly change the brightness, contrast, saturation and hue of an image. If the image is torch Tensor, it is expected to have […, 1 or 3, H, W] shape, where … means an arbitrary number of leading dimensions.