RandomGrayscale

class torchvision.transforms.RandomGrayscale(p=0.1)

以 p (預設為 0.1) 的機率隨機將圖像轉換為灰階。如果圖像為 torch Tensor，則預期具有 […, 3, H, W] 形狀，其中 … 表示任意數量的領先維度

參數:

p (float) – 圖像應轉換為灰階的機率。

傳回:

以機率 p 傳回輸入圖像的灰階版本，以機率 (1-p) 傳回未變更的版本。 - 如果輸入圖像為 1 個通道：灰階版本為 1 個通道 - 如果輸入圖像為 3 個通道：灰階版本為 3 個通道，其中 r == g == b

傳回類型:

PIL 圖像或 Tensor

forward(img)

參數:

img (PIL 圖像 或 Tensor) – 要轉換為灰階的圖像。

傳回:

隨機灰階化的圖像。

傳回類型:

PIL 圖像或 Tensor