快捷方式

RandomPosterize

class torchvision.transforms.v2.RandomPosterize(bits: int, p: float = 0.5)[原始碼]

以給定的機率,透過減少每個顏色通道的位數,將影像或影片進行色調分離 (posterize)。

如果輸入是 torch.Tensor,其型別應為 torch.uint8,且期望具有 […, 1 or 3, H, W] 的形狀,其中 … 表示任意數量的前導維度。如果 img 是 PIL Image,則期望其模式為 “L” 或 “RGB”。

引數:
  • bits (int) – 每個通道要保留的位數 (0-8)

  • p (float) – 影像被色調分離的機率。預設值為 0.5

使用 RandomPosterize 的示例

轉換示例

轉換示例
transform(inpt: Any, params: Dict[str, Any]) Any[原始碼]

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

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

文件

訪問 PyTorch 的全面開發者文件

檢視文件

教程

獲取針對初學者和高階開發者的深入教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源