equalize — Torchvision 0.22 documentation (original) (raw)

torchvision.transforms.functional.equalize(img: Tensor) → Tensor[source]¶

Equalize the histogram of an image by applying a non-linear mapping to the input in order to create a uniform distribution of grayscale values in the output.

Parameters:

img (PIL Image or Tensor) – Image on which equalize is applied. 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. The tensor dtype must be torch.uint8 and values are expected to be in [0, 255]. If img is PIL Image, it is expected to be in mode “P”, “L” or “RGB”.

Returns:

An image that was equalized.

Return type:

PIL Image or Tensor

Examples using equalize: