posterize¶
-
torchvision.transforms.functional.
posterize
(img: torch.Tensor, bits: int) → torch.Tensor[source]¶ Posterize an image by reducing the number of bits for each color channel.
- Parameters
img (PIL Image or Tensor) – Image to have its colors posterized. If img is torch Tensor, it should be of type torch.uint8 and 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”.
bits (int) – The number of bits to keep for each channel (0-8).
- Returns
Posterized image.
- Return type
PIL Image or Tensor
Examples using
posterize
: