RandomGrayscale

class torchvision.transforms.v2.RandomGrayscale(p: float = 0.1)

以機率 p (預設為 0.1) 隨機將影像或影片轉換為灰度圖。

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

輸出具有與輸入相同的通道數。

引數:

p (float) – 影像轉換為灰度的機率。

make_params(flat_inputs: List[Any]) → Dict[str, Any]

用於自定義變換的重寫方法。

參見 如何編寫自己的 v2 變換

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

用於自定義變換的重寫方法。

參見 如何編寫自己的 v2 變換