2

假设我有一个巨大的图像数据存储在 AWS S3 存储桶中。我需要使用 trochvision.datasets.ImageFolder 因为我的训练数据位于适当的文件夹中。所以我想在我的 Sagemaker 笔记本中使用以下代码。

from torch.utils.data import DataLoader
from torchvision import datasets, transforms

data_location = "s3_bucket/folder" 
batch_size= 16

transforms_= transforms.Compose([ transforms.RandomHorizontalFlip(p= 0.25),
                                transforms.RandomRotation(30),
                               transforms.RandomCrop(256, pad_if_needed=True),
                               transforms.ToTensor(),
                               transforms.Normalize([0.5, 0.5,0.5], [0.5,0.5,0.5]),
                               ])

train_dataset= datasets.ImageFolder(data_location, transform= transforms_)
train_loader= DataLoader(train_dataset, batch_size= batch_size, shuffle= True)

我怎样才能做到这一点？

4

0 回答 0