def load_data(batch_size,resize=None):
trans = [transforms.ToTensor()]
if resize:
trans.insert(0,transforms.Resize(resize))
trans = transforms.Compose(trans)
mnist_train = torchvision.datasets.FashionMNIST(
root="../data", train=True, transform=trans, download=False)
mnist_test = torchvision.datasets.FashionMNIST(
root="../data", train=False, transform=trans, download=False)
return data.DataLoader(mnist_train,batch_size),data.DataLoader(mnist_test,batch_size)
就是把多个transform类集成到一起了,像nn.Sequential()
本文介绍了一个用于加载FashionMNIST数据集的Python函数defload_data。该函数支持指定批次大小和图片尺寸调整,并使用torchvision进行数据处理。通过组合transforms操作,实现了数据标准化和尺寸调整的功能。
4274

被折叠的 条评论
为什么被折叠?



