快捷方式

灰度图

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

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

参数:

num_output_channels (int) – (1 或 3) 输出图像所需的通道数

返回值:

输入的灰度图版本。

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

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

返回类型:

PIL 图像

使用 Grayscale 的示例

变换的说明

变换的说明
forward(img)[source]
参数:

img (PIL ImageTensor) – 要转换为灰度图的图像。

返回值:

灰度图像。

返回类型:

PIL 图像或张量

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源