快捷方式

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}\]

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

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

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

  • gain (float) – 常数乘数。

返回:

伽马校正调整后的图像。

返回类型:

PIL 图像或 Tensor

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源