Shortcuts

Grayscale

class torchvision.transforms.Grayscale(num_output_channels=1)[source]

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:

forward(img)[source]
Parameters

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

Returns

Grayscaled image.

Return type

PIL Image or Tensor

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources