快捷方式

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 的示例

变换的图示

变换的图示

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

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

查看资源