solarize — Torchvision 0.22 documentation (original) (raw)

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: