Torch中怎么预处理数据

Torch

小亿

2024-03-07 18:53:30

栏目: 深度学习

在 Torch 中，可以使用 torchvision.transforms 模块来对数据进行预处理。常见的数据预处理操作包括：

图片缩放和裁剪：使用 transforms.Resize 和 transforms.CenterCrop。
图片翻转和旋转：使用 transforms.RandomHorizontalFlip 和 transforms.RandomRotation。
图片灰度化和归一化：使用 transforms.Grayscale 和 transforms.Normalize。
将图片转换为张量：使用 transforms.ToTensor。

以下是一个示例，展示如何使用 transforms 对数据进行预处理：

import torchvision.transforms as transforms

# 定义预处理操作
data_transform = transforms.Compose([
    transforms.Resize(256),  # 将图片缩放为 256x256
    transforms.CenterCrop(224),  # 将图片裁剪为 224x224
    transforms.RandomHorizontalFlip(),  # 随机水平翻转图片
    transforms.ToTensor(),  # 将图片转换为张量
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])  # 归一化图片像素值
])

# 加载数据集并应用预处理操作
train_dataset = torchvision.datasets.ImageFolder(root='path/to/train/data', transform=data_transform)

Torch中怎么预处理数据

最新问答

相关标签