datasets.MNIST(
root,
train=True,
transform=None,
target_transform=None,
download=False,
)
1.ROOT:数据集的根目录,数据地址
MNIST/processed/training.pt and MNIST/processed/test.pt exist
2.train (bool, optional):如果为True,则从training.pt创建数据集,
否则来自test.pt。
3.download (bool, optional): 如果为true,请从互联网下载数据集,然后
将其放在根目录中。 如果数据集已经下载,则不是再次下载。
4.transform:转换(可调用,可选):接受PIL图像并返回转换后版本的函数/转换。 例如,transforms.RandomCrop(就是对图像数据进行处理 传入dataloader的transform类型)
5.target_transform(可调用,可选):接受目标并对其进行转换的函数/转换。
datasets.MNIST()函数解析
最新推荐文章于 2023-11-01 21:10:52 发布