图像分类数据集

trans = transforms.ToTensor()
mnist_train = torchvision.datasets.FashionMNIST(
    root="../data", train=True, transform=trans, download=True)
mnist_test = torchvision.datasets.FashionMNIST(
    root="../data", train=False, transform=trans, download=True)

把图片换为tensor 

train = true 下载训练数据集

transform = trans 下载的是tensor 不是图片格式

 当我们构建好数据集后 可以放进dataloader里面

我们可以指定batch-size，构造一个iter ，next 就是拿到第一个小批量

X, y = next(iter(data.DataLoader(mnist_train, batch_size=18)))
show_images(X.reshape(18, 28, 28), 2, 9, titles=get_fashion_mnist_labels(y));