Shortcuts

adjust_gamma

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

对图像执行伽马校正。

也被称为幂律变换。RGB模式下的强度根据以下公式进行调整:

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

有关更多详细信息,请参见Gamma Correction

Parameters:
  • img (PIL ImageTensor) – 需要调整的PIL图像。 如果img是torch张量,则期望其格式为[…, 1或3, H, W], 其中…表示它可以有任意数量的前导维度。 如果img是PIL图像,则不支持具有透明度(alpha通道)的模式。

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

  • gain (float) – 常数乘数。

Returns:

伽马校正调整后的图像。

Return type:

PIL 图像或张量