- 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
num_output_channels (int) – (1 or 3) number of channels desired for output image
Grayscale version of the input.
num_output_channels == 1: returned image is single channel
num_output_channels == 3: returned image is 3 channel with r == g == b
- Return type:
Grayscale:Illustration of transforms