快捷方式

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\) 相同。gamma 大于 1 使阴影变暗,而 gamma 小于 1 使暗区变亮。

  • gain (float) – 常数乘数。

返回:

伽马校正后的图像。

返回类型:

PIL Image 或 Tensor

文档

查阅全面的 PyTorch 开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源