快捷方式

to_grayscale

torchvision.transforms.functional.to_grayscale(img, num_output_channels=1)[source]

将任何模式(RGB、HSV、LAB 等)的 PIL 图像转换为灰度版本。此转换不支持 torch Tensor。

参数:
  • img (PIL Image) – 要转换为灰度的 PIL 图像。

  • num_output_channels (int) – 输出图像的通道数。值可以是 1 或 3。默认为 1。

返回:

图像的灰度版本。

  • 如果 num_output_channels = 1:返回的图像是单通道

  • 如果 num_output_channels = 3:返回的图像是 3 通道,其中 r = g = b

返回类型:

PIL Image

使用 to_grayscale 的示例

转换图示

转换图示

文档

访问全面的 PyTorch 开发者文档

查看文档

教程

为初学者和高级开发者提供深入的教程

查看教程

资源

查找开发资源并让您的问题得到解答

查看资源