RandomSolarize

class torchvision.transforms.v2.RandomSolarize(threshold: float, p: float = 0.5)

以给定的概率对图像或视频进行 Solarize 处理，方法是反转高于阈值的所有像素值。

如果 img 是 Tensor，则应为 […, 1 或 3, H, W] 格式，其中 … 表示它可以具有任意数量的前导维度。如果 img 是 PIL 图像，则应为 “L” 或 “RGB” 模式。

参数:

• threshold (float) – 所有大于或等于此值的像素都会被反转。

• p (float) – 图像被 Solarize 处理的概率。默认值为 0.5

使用 RandomSolarize 的示例

转换的图示

转换的图示

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

用于覆盖自定义转换的方法。

参见 如何编写您自己的 v2 转换