RandomGrayscale

class torchvision.transforms.v2.RandomGrayscale(p: float = 0.1)

以 p（默认为 0.1）的概率随机将图像或视频转换为灰度。

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

输出具有与输入相同的通道数。

参数:

p (float) – 图像应转换为灰度的概率。

make_params(flat_inputs: list[Any]) → dict[str, Any]

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

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

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

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

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