datasets.MNIST()函数解析

dedededer

于 2020-04-19 16:19:21 发布

阅读量1.3w

点赞数

文章标签： pytorch

本文链接：https://blog.youkuaiyun.com/michaelchan1234/article/details/105616525

版权

本文介绍了PyTorch中datasets.MNIST()函数的使用，包括root参数指定数据集根目录，train参数选择训练集或测试集，transform参数用于图像预处理，download参数决定是否自动下载数据集。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

datasets.MNIST(
root,
train=True,
transform=None,
target_transform=None,
download=False,
)
Docstring:
MNIST <http://yann.lecun.com/exdb/mnist/>_ Dataset.

Args:
root (string): Root directory of dataset where MNIST/processed/training.pt
and MNIST/processed/test.pt exist.
train (bool, optional): If True, creates dataset from training.pt,
otherwise from test.pt.
download (bool, optional): If true, downloads the dataset from the internet and
puts it in root directory. If dataset is already downloaded, it is not
downloaded again.
transform (callable, optional): A function/transform that takes in an PIL image
and returns a transformed version. E.g, transforms.RandomCrop
target_transform (callable, optional): A function/transform that takes in the
target and transforms it.

最低0.47元/天解锁文章