随机灰度

class torchvision.transforms.RandomGrayscale(p=0.1)

以 p（默认值为 0.1）的概率随机将图像转换为灰度。如果图像为 torch 张量，则预计它将具有 […, 3, H, W] 形状，其中 … 表示任意数量的领先维度

参数：

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

返回值：

以 p 的概率为灰度版本，以 (1-p) 的概率保持不变的输入图像的灰度版本。- 如果输入图像为 1 通道：灰度版本为 1 通道 - 如果输入图像为 3 通道：灰度版本为 3 通道，其中 r == g == b

返回类型：

PIL 图像或张量

forward(img)

参数：

img (PIL 图像 或 张量) – 要转换为灰度的图像。

返回值：

随机灰度化的图像。

返回类型：

PIL 图像或张量