【PyTorch】torchvision进行图像预处理

最新推荐文章于 2025-04-10 19:37:35 发布

颜良文良

最新推荐文章于 2025-04-10 19:37:35 发布

阅读量1.9k

点赞数

分类专栏： PyTorch 文章标签： PyTorch

本文链接：https://blog.youkuaiyun.com/bigFatCat_Tom/article/details/99464811

版权

PyTorch 专栏收录该内容

21 篇文章

订阅专栏

本文介绍使用PyTorch进行图像数据预处理的方法，包括随机裁剪、水平翻转、转换为张量及标准化等步骤，以提高深度学习模型训练效果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import torch
from torchvision import transforms, datasets

data_transform = transforms.Compose([
        transforms.RandomSizedCrop(224),
        transforms.RandomHorizontalFlip(),
        transforms.ToTensor(),
        transforms.Normalize(mean=[0.485, 0.456, 0.406],
                             std=[0.229, 0.224, 0.225])
    ])
hymenoptera_dataset = datasets.ImageFolder(root='hymenoptera_data/train',
                                           transform=data_transform)
dataset_loader = torch.utils.data.DataLoader(hymenoptera_dataset,
                                             batch_size=4, shuffle=True,
                                             num_workers=4)