Convert image to grayscale — transform_grayscale • torchvision

Convert image to grayscale

transform_grayscale(img, num_output_channels)

Arguments

img: A magick-image, array or torch_tensor.
num_output_channels: (int): (1 or 3) number of channels desired for output image

See also

Other unitary_transforms: transform_adjust_brightness(), transform_adjust_contrast(), transform_adjust_gamma(), transform_adjust_hue(), transform_adjust_saturation(), transform_affine(), transform_center_crop(), transform_convert_image_dtype(), transform_crop(), transform_hflip(), transform_linear_transformation(), transform_normalize(), transform_pad(), transform_perspective(), transform_resize(), transform_rgb_to_grayscale(), transform_rotate(), transform_to_tensor(), transform_vflip()