数据集_FashionMNIST 数据集

FashionMNIST数据集

本文主要介绍FashionMNIST数据集，可以在这里查看FashionMNIST数据集更详细的内容
FashionMNIST。

在做图像分类任务中，MNIST数据集很容易就能够取得很好的分类效果，因此可以尝试使用FashionMNIST数据集来验证模型。

接下来介绍如何下载FashionMNIST数据集。

下载时注意修改下载路径。

import torch
import torchvision
import torchvision.transforms as transformers
import sys
import matplotlib.pyplot as plt
import time
# 使用pycharm时注意加
import os
os.environ['KMP_DUPLICATE_LIB_OK'] = 'True'
mnist_train = torchvision.datasets.FashionMNIST(root=r'G:\Desktop\Pytorch_image_classification\Project01_FashionMNIST\dataset', 
                                                train=True, download=True, transform=transformers.ToTensor())
mnist_test  = torchvision.datasets.FashionMNIST(root=r'G:\Desktop\Pytorch_image_classification\Project01_FashionMNIST\dataset', 
                                                train=False, download=True, transform=transformers.ToTensor())

print(type(mnist_train))
print(len(mnist_train), len(mnist_test))

输出结果为：

<class 'torchvision.datasets.mnist.FashionMNIST'>
60000 10000

至此，已经将FashionMNIST数据集下载好了。