快捷方式

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 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题解答

查看资源