灰階

class torchvision.transforms.v2.Grayscale(num_output_channels: int = 1)

將圖像或影片轉換為灰階。

如果輸入為 torch.Tensor，則預期具有 […, 3 或 1, H, W] 形狀，其中 … 表示任意數量的先導維度

參數:

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

使用 Grayscale 的範例

轉換的圖示

轉換的圖示

transform(inpt: Any, params: Dict[str, Any]) → Any

覆寫自訂轉換的方法。

請參閱 如何編寫您自己的 v2 轉換