如何將輸入和目標與 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]))