RandomPosterize

class torchvision.transforms.RandomPosterize(bits, p=0.5)

Posterize the image randomly with a given probability by reducing the number of bits for each color channel. If the image is torch Tensor, it should be of type torch.uint8, and it is expected to have […, 1 or 3, H, W] shape, where … means an arbitrary number of leading dimensions. If img is PIL Image, it is expected to be in mode “L” or “RGB”.

Parameters

• bits (int) – number of bits to keep for each channel (0-8)

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

Examples using RandomPosterize:

Illustration of transforms

Illustration of transforms

forward(img)

Parameters

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

Returns

Randomly posterized image.

Return type

PIL Image or Tensor