solarize¶
- torchvision.transforms.functional.solarize(img: Tensor, threshold: float) Tensor[source]¶
Solarize an RGB/grayscale image by inverting all pixel values above a threshold.
- Parameters:
img (PIL Image or Tensor) – Image to have its colors inverted. If img is torch 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”.
threshold (float) – All pixels equal or above this value are inverted.
- Returns:
Solarized image.
- Return type:
PIL Image or Tensor
Examples using
solarize:
