pytorch中的transforms模块实例详解

在PyTorch中，transforms模块提供了一系列用于数据预处理和数据增强的函数。以下是两个示例说明。

示例1：使用transforms进行数据预处理

import torch
import torchvision
import torchvision.transforms as transforms

# 定义transforms
transform = transforms.Compose(
    [transforms.ToTensor(),
     transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])

# 加载数据集
trainset = torchvision.datasets.CIFAR10(root='./data', train=True,
                                        download=True, transform=transform)
trainloader = torch.utils.data.DataLoader(trainset, batch_size=4,
                                          shuffle=True, num_workers=2)

# 输出数据集
dataiter = iter(trainloader)
images, labels = dataiter.next()
print(images.shape)

在这个示例中，我们首先定义了一个名为transform的Compose对象，其中包含了两个预处理函数：ToTensor和Normalize。然后，我们使用torchvision.datasets.CIFAR10函数加载CIFAR10数据集，并将transform对象传递给transform参数。最后，我们使用torch.utils.data.DataLoader函数加载数据集，并使用iter函数和next函数获取一个batch的数据。

示例2：使用transforms进行数据增强

import torch
import torchvision
import torchvision.transforms as transforms

# 定义transforms
transform = transforms.Compose(
    [transforms.RandomCrop(32, padding=4),
     transforms.RandomHorizontalFlip(),
     transforms.ToTensor(),
     transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])

# 加载数据集
trainset = torchvision.datasets.CIFAR10(root='./data', train=True,
                                        download=True, transform=transform)
trainloader = torch.utils.data.DataLoader(trainset, batch_size=4,
                                          shuffle=True, num_workers=2)

# 输出数据集
dataiter = iter(trainloader)
images, labels = dataiter.next()
print(images.shape)

在这个示例中，我们首先定义了一个名为transform的Compose对象，其中包含了三个数据增强函数：RandomCrop、RandomHorizontalFlip和ToTensor。然后，我们使用torchvision.datasets.CIFAR10函数加载CIFAR10数据集，并将transform对象传递给transform参数。最后，我们使用torch.utils.data.DataLoader函数加载数据集，并使用iter函数和next函数获取一个batch的数据。