pillow中ImageFolder()详解

Pillow是Python中一个非常强大的图像处理库，可以对图像进行各种操作。其中，ImageFolder()是Pillow中的一个函数，用于读取文件夹中的所有图像文件，将其加载为一个数据集。

具体来说，ImageFolder()函数可以接受两个参数：文件夹路径和一个可选的变换函数。文件夹路径指定了包含图像文件的文件夹，变换函数可以对图像进行一些转换，比如裁剪、缩放等。

ImageFolder()函数返回一个数据集，其中每个元素都是一个图像和其对应的标签。标签是从文件夹名字中提取的，所以需要满足一定的命名规范。

下面是一个使用ImageFolder()函数读取数据集的示例代码：

from torchvision import datasets, transforms

# 定义变换函数
transform = transforms.Compose([
    transforms.RandomResizedCrop(224),
    transforms.RandomHorizontalFlip(),
    transforms.ToTensor(),
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
])

# 加载数据集
dataset = datasets.ImageFolder('data', transform=transform)

# 获取数据集大小和类别数
print('Dataset size:', len(dataset))
print('Number of classes:', len(dataset.classes))

在上面的代码中，我们首先定义了一个变换函数，其中包含了随机裁剪、随机水平翻转、归一化等操作。然后使用ImageFolder()函数读取data文件夹中的图像文件，并应用变换函数，得到一个数据集。最后，我们输出了数据集的大小和类别数，分别是200张图像和2个类别。