torchvision.datasets中包含如下数据集:
dset.MNIST(root, train=True, transform=None, target_transform=None, download=False)
参数说明:
processed/training.pt
和 processed/test.pt
的主目录一个通用的数据加载器,数据集中的数据以以下方式组织
root/dog/xxx.png
root/dog/xxy.png
root/dog/xxz.png
root/cat/123.png
root/cat/nsdf3.png
root/cat/asd932_.png
dset.ImageFolder(root="root folder path", [transform, target_transform])
有以下成员变量:
root是根文件夹目录
At the heart of PyTorch data loading utility is the torch.utils.data.DataLoader class. It represents a Python iterable over a dataset, with support for
pytorch数据加载的核心是torch.utils.data.DataLoader类,它表示在一个数据集上的Python迭代。
DataLoader(dataset, batch_size=1, shuffle=False, sampler=None,
batch_sampler=None, num_workers=0, collate_fn=None,
pin_memory=False, drop_last=False, timeout=0,
worker_init_fn=None, *, prefetch_factor=2,
persistent_workers=False)
参数说明: