随机灰度

class torchvision.transforms.RandomGrayscale(p=0.1)

以概率p（默认0.1）随机将图像转换为灰度图像。 如果图像是torch Tensor，则期望其具有[…, 3, H, W]形状，其中…表示任意数量的前导维度

Parameters:

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

Returns:

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

Return type:

PIL 图像或张量

forward(img)

Parameters:

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

Returns:

随机灰度化的图像。

Return type:

PIL 图像或张量