RandomInvert
- class torchvision.transforms.RandomInvert(p=0.5)[source]
Inverts the colors of the given image randomly with a given probability. If img is a Tensor, it is expected to be in […, 1 or 3, H, W] format, where … means it can have an arbitrary number of leading dimensions. If img is PIL Image, it is expected to be in mode “L” or “RGB”.
- Parameters:
p (float) – probability of the image being color inverted. Default value is 0.5
Examples using
RandomInvert
:- forward(img)[source]
- Parameters:
img (PIL Image or Tensor) – Image to be inverted.
- Returns:
Randomly color inverted image.
- Return type:
PIL Image or Tensor