Grayscale

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 變換