快捷方式

rgb_to_grayscale

torchvision.transforms.functional.rgb_to_grayscale(img: Tensor, num_output_channels: int = 1) Tensor[源代码]

将 RGB 图像转换为图像的灰度版本。如果图像为 torch 张量,则预计其形状为 […, 3, H, W],其中 … 表示任意数量的前导维度。

注意

请注意,此方法仅支持 RGB 图像作为输入。对于其他颜色空间中的输入,请考虑使用 to_grayscale() 和 PIL Image。

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

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

返回值:

图像的灰度版本。

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

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

返回类型:

PIL Image 或 Tensor

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源