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.
Resize — Torchvision main documentation
pytorch.org › generated › torchvisionResize. class torchvision.transforms.Resize(size, interpolation=<InterpolationMode.BILINEAR: 'bilinear'>, max_size=None, antialias=None) [source] Resize the input image to the given size. If the image is torch Tensor, it is expected to have […, H, W] shape, where … means an arbitrary number of leading dimensions. Warning.