Grayscale

class torchvision.transforms.Grayscale(num_output_channels=1)

Convert image to grayscale. If the image is torch Tensor, it is expected to have […, 3, H, W] shape, where … means an arbitrary number of leading dimensions

Parameters

num_output_channels (int) – (1 or 3) number of channels desired for output image

Returns

Grayscale version of the input.

• If num_output_channels == 1 : returned image is single channel

• If num_output_channels == 3 : returned image is 3 channel with r == g == b

Return type

PIL Image

Examples using Grayscale:

Illustration of transforms

Illustration of transforms

forward(img)

Parameters

img (PIL Image or Tensor) – Image to be converted to grayscale.

Returns

Grayscaled image.

Return type

PIL Image or Tensor