如何在 PyTorch 中调整图像的对比度？

  • Z时代
  • 2024-01-10
  • 分类：综合

图像的对比度是指图像的各种特征之间存在的颜色差异量。为了调整图像的对比度，我们应用adjust_contrast()。它是torchvision.transforms模块提供的功能转换之一。该模块包含许多重要的功能转换，可用于对图像数据执行不同类型的操作。

adjust_contrast()转换接受 PIL 和张量图像。张量图像是形状为[C, H, W]的 PyTorch 张量，其中C是通道数，H是图像高度，W是图像宽度。此变换还接受一批张量图像，它是具有[B, C, H, W]的张量，其中B是批次中的图像数量。如果图像既不是 PIL 图像也不是张量图像，那么我们首先将其转换为张量图像，然后应用adjust_contrast(). 所有功能转换都可以从torchvision.transforms.functional 访问。

语法

torchvision.transforms.functional.adjust_contrast(img, contrast_factor)

参数

  • img - 这是要调整对比度的图像。它是 PIL 图像或火炬张量。它可能是单个图像或一批图像。

  • contrast_factor - 一个非负数。0 给出纯灰色图像，1 给出原始图像。

输出结果

它返回对比度调整后的图像。

脚步

要调整图像的对比度，可以按照以下步骤进行

  • 导入所需的库。在以下所有示例中，所需的 Python 库是torch.、Pillow和torchvision确保您已经安装了它们。

import torch
import torchvision
import torchvision.transforms.functional as F
from PIL import Image

  • 读取输入图像。输入图像是 PIL 图像或 Torch 张量。

img = Image.open('Nature_canada.jpg')

  • 使用所需的对比度系数调整图像的对比度。

img = F.adjust_contrast(img, 0.3)

  • 可视化对比度调整后的图像。

img.show()

输入图像

在以下示例中，我们将使用此图像作为输入文件

示例 1

在下面的 Python 程序中，我们使用contrast_factor=0.3调整输入图像的对比度。

# Import the required libraries
import torch
from PIL import Image
import torchvision.transforms.functional as F
# read the input image
img = Image.open('Nature_canada.jpg')
# adjust the contrast of the image
img = F.adjust_contrast(img, 0.3)
# display the contrast adjusted image
img.show()
输出结果

示例 2

在下面的 Python 程序中，我们使用contrast_factor=6 调整输入图像的对比度。

import torch
from PIL import Image
import torchvision.transforms.functional as F
img = Image.open('Nature_canada.jpg')
img = F.adjust_contrast(img, 6)
img.show()
输出结果

以上是 如何在 PyTorch 中调整图像的对比度？ 的全部内容， 来源链接： utcz.com/z/363472.html

回到顶部