torchvision.transforms — Torchvision master documentation
https://pytorch.org/vision/0.9/transforms.htmlclass 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 […, 3, H, W] shape, where … means an arbitrary number of …