灰度

class torchvision.transforms.v2.Grayscale(num_output_channels: int = 1)

将图像或视频转换为灰度。

如果输入是torch.Tensor，则期望其具有[…, 3 或 1, H, W]形状，其中…表示任意数量的前导维度

Parameters:

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

使用Grayscale的示例：

变换的图示

Illustration of transforms