RandomSolarize — Torchvision 0.22 documentation (original) (raw)

class torchvision.transforms.RandomSolarize(threshold, p=0.5)[source]¶

Solarize the image randomly 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:

forward(img)[source]¶

Parameters:

img (PIL Image or Tensor) – Image to be solarized.

Returns:

Randomly solarized image.

Return type:

PIL Image or Tensor