快捷方式

adjust_gamma

torchvision.transforms.functional.adjust_gamma(img: Tensor, gamma: float, gain: float = 1) Tensor[源代码]

对图像执行伽马校正。

也称为幂律变换。RGB 模式下的亮度根据以下方程进行调整

\[I_{\text{out}} = 255 \times \text{gain} \times \left(\frac{I_{\text{in}}}{255}\right)^{\gamma}\]

有关更多详细信息,请参阅 伽马校正

参数:
  • img (PIL ImageTensor) – 要调整的 PIL Image。如果 img 是 torch Tensor,则应为 […, 1 或 3, H, W] 格式,其中 … 表示它可以具有任意数量的前导维度。如果 img 是 PIL Image,则不支持带透明度(Alpha 通道)的模式。

  • gamma (float) – 非负实数,与方程中的 \(\gamma\) 相同。大于 1 的 gamma 使阴影变暗,而小于 1 的 gamma 使暗区变亮。

  • gain (float) – 常数乘数。

返回:

伽马校正调整后的图像。

返回类型:

PIL 图像或张量

文档

访问全面的 PyTorch 开发者文档

查看文档

教程

为初学者和高级开发者提供深入的教程

查看教程

资源

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

查看资源