Shortcuts

RandomSolarize

class torchvision.transforms.v2.RandomSolarize(threshold: float, p: float = 0.5)[source]

Solarize the image or video with a given probability by inverting all pixel values above a threshold.

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:
  • threshold (float) – all pixels equal or above this value are inverted.

  • p (float) – probability of the image being solarized. Default value is 0.5

Examples using RandomSolarize:

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