快捷方式

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 图像张量) – 要调整的 PIL 图像。如果 img 是 torch 张量,则预期其格式为 […, 1 或 3, H, W],其中 … 表示它可以具有任意数量的前导维度。如果 img 是 PIL 图像,则不支持具有透明度(alpha 通道)的模式。

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

  • gain (浮点数) – 常数乘数。

返回值:

伽马校正调整后的图像。

返回类型:

PIL 图像或张量

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源