灰度

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

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

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

参数:

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

使用 Grayscale 的示例

变换示例

变换示例

transform(inpt: Any, params: Dict[str, Any]) → Any

用于覆盖以实现自定义变换的方法。

请参阅 如何编写你自己的 v2 变换