动手深度学习6-认识Fashion_MNIST图像数据集
本节将使用torchvision包,它是服务于pytorch深度学习框架的,主要用来构建计算机视觉模型。 torchvision主要由以下几个部分构成: torchvision.datasets:一些加载数据的函数以及常用的数据集的接口 torchvision.models: 包含常用的模型结构(含预训练模型),例如AlexNet,VGG,ResNet; torchvision.transforms:常用的图片变换,例如裁剪,旋转等; torchvision.utils: 其他的一些有用的方法 导入本节需要的包或者模块 import torch import torchvision import torchvision.transforms as transforms import matplotlib.pyplot as plt import time import sys sys.path.append('..') # 为了导入上层目录的d2lzh_pytorch import d2lzh_pytorch as d2l 通过调用torchvision的torchvision.datasets来下载这个数据集 可以通过train参数获取指定的训练集或者测试集、 测试集只用了评估模型,并不用来训练模型 同时指定了参数transform = transform.ToTensor(