快捷方式

solarize

torchvision.transforms.functional.solarize(img: Tensor, threshold: float) Tensor[源代码]

通过反转高于阈值的所有像素值来使 RGB/灰度图像太阳化。

参数:
  • img (PIL ImageTensor) – 需要反转颜色的图像。如果 img 是 torch 张量,则预计其格式为 […, 1 或 3, H, W],其中 … 表示它可以具有任意数量的前导维度。如果 img 是 PIL Image,则预计其模式为“L”或“RGB”。

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

返回值:

太阳化的图像。

返回类型:

PIL Image 或 Tensor

使用 solarize 的示例

转换图示

转换图示

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

查找开发资源并解答您的问题

查看资源