Shortcuts

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:

Illustration of transforms

Illustration of transforms

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources