solarize¶
-
torchvision.transforms.functional.
solarize
(img: torch.Tensor, threshold: float) → torch.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
: