快捷方式

灰度

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

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

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

参数:

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

使用 Grayscale 的示例

变换说明

变换说明

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源