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