Grayscale

class torchaug.transforms.Grayscale(num_output_channels=1, batch_transform=False)

Convert images or videos to grayscale.

If the input is a torch.Tensor, it is expected to have […, 3 or 1, H, W] shape, where … means an arbitrary number of leading dimensions

Parameters:

• num_output_channels (int, optional) – (1 or 3) number of channels desired for output image. Default: 1

• batch_transform (bool, optional) – whether to apply the transform in batch mode. Default: False

extra_repr()

Set the extra representation of the transform.

Return type:

str