torchvision.datasets.FashionMNIST
是 PyTorch 的一个数据集类,用于加载 Fashion MNIST 数据集,这是一个经典的用于图像分类的基准数据集。这个数据集包含了 60,000 张训练图片和 10,000 张测试图片,每张图片都是 28x28 像素的灰度图,属于 10 个类别中的一种(例如 T 恤、裤子、鞋子等)。
-
torchvision.datasets.FashionMNIST
返回的数据类型:train_dataset
是一个torchvision.datasets.FashionMNIST
对象。- 该对象继承自
torch.utils.data.Dataset
,因此可以用来索引和迭代。 - 每个元素是一个二元组
(image, label)
:image
是一个PIL
图像对象(默认情况下)或者是经过transform
变换后的张量。label
是一个整数,表示图像的类别(从 0 到 9)。