如果你是AI零基础,请关注本专栏,将带你一起飞。
在PyTorch中,通常使用torchvision和torch.utils.data模块实现数据集的加载功能。在此模块中提供了用于加载和预处理常见数据集的工具,同时也支持自定义数据集的加载。
2.1.1 PyTorch加载数据集
在PyTorch程序中,模块torchvision.datasets提供了许多常见的预定义数据集,并提供了简单的API来加载这些数据集。以下是一些常用的数据集加载函数:
- torchvision.datasets.ImageFolder:用于加载图像文件夹数据集,其中每个子文件夹表示一个类别,文件夹中的图像属于该类别。
- torchvision.datasets.CIFAR10和torchvision.datasets.CIFAR100:用于加载CIFAR-10和CIFAR-100数据集,这是两个广泛使用的图像分类数据集。
- torchvision.datasets.MNIST:用于加载MNIST手写数字数据集,其中包含了大量的手写数字图像及其对应的标签。
- torchvision