torchvision.transforms — Torchvision 0.11.0 documentation
pytorch.org › vision › stableclass 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.