solarize¶
- torchvision.transforms.functional.solarize(img: Tensor, threshold: float) Tensor [source]¶
通过反转阈值以上的所有像素值来对 RGB/灰度图像进行 solarize 处理。
- 参数:
img (PIL 图像 或 Tensor) – 要反转颜色的图像。如果 img 是 torch Tensor,则应为 […, 1 或 3, H, W] 格式,其中 … 表示它可以具有任意数量的前导维度。如果 img 是 PIL 图像,则应为 “L” 或 “RGB” 模式。
threshold (float) – 所有大于或等于此值的像素将被反转。
- 返回:
Solarize 处理后的图像。
- 返回类型:
PIL 图像或 Tensor
使用
solarize
的示例