如何将输入和目标与 Pytorch Fashion MNIST 分开？

Question

Fashion MNIST 数据集在 Pytorch 中的实现非常奇怪。我想做类似的事情：

X, y = FashionMNIST

但实际上，情况要复杂一些。这就是我所拥有的：

from torchvision.datasets import FashionMNIST
train = FashionMNIST(root='.', download=True, train=True)
print(train)

输出：

Dataset FashionMNIST
    Number of datapoints: 60000
    Root location: c:/users/nicolas/documents/data/fashionmnist
    Split: Train

什么一个观察的样子：

print(train[0])

(<PIL.Image.Image image mode=L size=28x28 at 0x20868074780>, 9)

我只能做一次观察。

X, y = train[0]

那么如何分离输入和目标呢？

Answer 1

FashionMNIST对象具有data和targets属性。

你可以简单地写

X, y = train.data, train.targets

然后你可以看到形状

X.shape, y.shape

(torch.Size([60000, 28, 28]), torch.Size([60000]))