Pytorch基础及实战(4)——transforms数据增强

在AI领域的模型训练中通常会遇到模型过拟合问题，通常采取的办法就是数据增强处理，例如在图像处理中，数据增强是指对原始图像进行旋转、缩放、剪切、翻转等操作，以扩大训练数据集的规模，提高模型泛化能力，降低过拟合风险。

笔者在这里以深度学习框架Pytorch中的数据增强工具(transforms模块)为例介绍数据增强处理。torchvision.transforms是PyTorch中用于图像处理和数据增强的模块。它提供了许多函数，可以在图像上应用各种转换，例如裁剪、旋转、翻转、缩放、归一化等操作，从而生成更多变化的图像数据。

transforms模块中的函数可以分为两类：一类是针对PIL图像对象的操作函数，例如：Resize、RandomCrop、RandomHorizontalFlip等；另一类是对Tensor对象的操作函数，例如：Normalize、ToTensor等。下面笔者将分别介绍这些函数的原理和代码示例。

针对PIL图像对象的操作函数

Resize

Resize函数可以将图像缩放到指定大小。常用的缩放方法等比例缩放和非等比例缩放两种。其中如果预先目标大小，Resize函数会按照目标大小等比例缩放图像，如果预先仅指定了图片宽度或高度，则会进行非等比例缩放。

from torchvision import transforms
from PIL import Image

# 等比例缩放
transform = transforms.Compose([transforms.Resize((224, 224))])
img = Image.open('1.png')
img.show(img)
img = transform(img)
img.show(img)

# # 非等比例缩放
transform = transforms.Compose([transforms.Resize((224, 300))])
img = Image.open('1.png')
img.show(img)
img = transform(img)
img.show(img)

• 等比例缩放
  

• 非等比例缩放

CenterCrop

CenterCrop函数可以从图像中心裁剪指定大小的区域。

from torchvision import transforms
from PIL import Image

transform = transforms.Compose([transforms.CenterCrop(224)])
img = Image.open('1.png')
img.show(img)
img = transform(img)
img.show(img)

RandomCrop

RandomCrop函数可以随机裁剪指定大小的区域。

from torchvision import transforms
from PIL import Image

transform = transforms.Compose([transforms.RandomCrop(224)])
img = Image.open('1.png')
img.show(img)
img = transform(img)
img.show(img)

RandomHorizontalFlip

RandomHorizontalFlip函数可以随机水平翻转图像。

from torchvision import transforms
from PIL import Image

transform = transforms.Compose([transforms.RandomHorizontalFlip()])
img = Image.open('1.png')
img.show(img)
img = transform(img)
img.show(img)

针对Tensor对象的操作函数

ToTensor

ToTensor函数可以将PIL图像对象转换为Tensor对象。

from torchvision import transforms
from PIL import Image

transform = transforms.Compose([transforms.ToTensor()])
img = Image.open('1.png')

img = transform(img)
print(img.size()) #torch.Size([3, 800, 1000])

Normalize

Normalize函数可以对Tensor对象进行归一化，以减少模型训练的时间，提高模型性能和稳定性。

from torchvision import transforms
from PIL import Image
import torch

transform = transforms.Compose([transforms.ToTensor(),
                                transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])
img = Image.open('1.png')
img = transform(img)
print(torch.min(img), torch.max(img)) # tensor(-1.) tensor(1.)

上述代码将图像像素值归一化到[-1, 1]之间。

Compose函数

Compose函数则用于将多个transforms函数组合在一起，形成一个transforms的列表。在数据加载时，会按照列表中的顺序，依次对图像进行变换。

from torchvision import transforms

transform = transforms.Compose([
    transforms.CenterCrop(224),
    transforms.RandomHorizontalFlip(),
    transforms.ToTensor(),
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
])
img = Image.open('1.png')
img = transform(img)
print(torch.min(img), torch.max(img))

上述代码中，我们通过Compose函数将CenterCrop、RandomHorizontalFlip、ToTensor和Normalize四个函数组合在一起，形成了一个transform对象。