如何将输入和目标与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]))