快捷方式

RandomGrayscale

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

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

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

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

參數:

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

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

覆寫自訂轉換的方法。

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

transform(inpt: Any, params: Dict[str, Any]) Any[原始碼]

覆寫自訂轉換的方法。

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

文件

取得 PyTorch 完整的開發者文件

檢視文件

教學

取得針對初學者和進階開發者的深入教學

檢視教學

資源

尋找開發資源並獲得問題解答

檢視資源