阅读量:1
在PyTorch中，可以使用torchvision.transforms来实现数据增强。torchvision.transforms提供了一系列用于对图像进行数据增强的函数，可以在数据加载时对图像进行随机裁剪、翻转、旋转、缩放等操作。
以下是一个示例代码，演示了如何在PyTorch中使用torchvision.transforms对图像进行数据增强：
import torch from torchvision import datasets, transforms # 定义数据增强的操作 transform = transforms.Compose([ transforms.RandomHorizontalFlip(), # 随机水平翻转 transforms.RandomRotation(10), # 随机旋转角度 transforms.ColorJitter(brightness=0.2, contrast=0.2, saturation=0.2, hue=0.2), # 随机调整颜色 transforms.RandomResizedCrop(224), # 随机裁剪并缩放 transforms.ToTensor() # 转换为Tensor ]) # 加载数据集并应用数据增强 train_dataset = datasets.CIFAR10(root='./data', train=True, download=True, transform=transform) train_loader = torch.utils.data.DataLoader(train_dataset, batch_size=32, shuffle=True) # 使用train_loader进行训练 for images, labels in train_loader: # 进行训练 pass 在上面的示例中，定义了一个包含多种数据增强操作的transform，并将其应用在CIFAR10数据集上。然后使用torch.utils.data.DataLoader加载数据集，并传入transform参数，从而在训练过程中对图像进行数据增强。
