torchvision.tranforms 解析

最新推荐文章于 2024-10-02 19:08:53 发布

yjinyyzyq

最新推荐文章于 2024-10-02 19:08:53 发布

阅读量193

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/yjinyyzyq/article/details/90693337

博客对代码进行了概览与解析，主要涉及torchvison.transforms.Compose函数，以及其中a为torchvision.tranforms.Resize((32, 32))的设置，属于信息技术中深度学习相关代码内容。

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

torchvision.transforms

代码概览
torchvison.transforms.Compose([a, b, c, d, e, f])
a = torchvision.tranforms.Resize((32, 32))

代码概览

import torch
from torchvision import transforms, datasets

data_transform = transforms.Compose([
        transforms.RandomSizedCrop(224),
        transforms.Resize((32, 32))
        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)