灰階

class torchvision.transforms.Grayscale(num_output_channels=1)

將圖片轉換為灰階。如果圖片是 torch Tensor，則預期具有 […, 3, H, W] 形狀，其中 … 表示任意數量的領先維度

參數:

num_output_channels (int) – (1 或 3) 輸出圖片所需的通道數

回傳:

輸入的灰階版本。

• 如果 num_output_channels == 1 : 回傳的圖片是單一通道

• 如果 num_output_channels == 3 : 回傳的圖片是 3 個通道，且 r == g == b

回傳類型:

PIL Image

使用 Grayscale 的範例

轉換的圖示

轉換的圖示

forward(img)

參數:

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

回傳:

灰階圖片。

回傳類型:

PIL Image 或 Tensor