RandomGrayscale

class torchvision.transforms.RandomGrayscale(p=0.1)

Randomly convert image to grayscale with a probability of p (default 0.1). If the image is torch Tensor, it is expected to have […, 3, H, W] shape, where … means an arbitrary number of leading dimensions

Parameters:

p (float) – probability that image should be converted to grayscale.

Returns:

Grayscale version of the input image with probability p and unchanged with probability (1-p). - If input image is 1 channel: grayscale version is 1 channel - If input image is 3 channel: grayscale version is 3 channel with r == g == b

Return type:

PIL Image or Tensor

forward(img)

Parameters:

img (PIL Image or Tensor) – Image to be converted to grayscale.

Returns:

Randomly grayscaled image.

Return type:

PIL Image or Tensor