ImageFolder函数用法

`ImageFolder`是PyTorch中的一个数据加载器，用于从文件夹中加载图像数据集。它假定数据集的每个类别都存储在单独的文件夹中，并且每个文件夹的名称为该类别的名称。该函数的用法如下：

torchvision.datasets.ImageFolder(root, transform=None, target_transform=None, loader=<function default_loader>, is_valid_file=None)

参数说明：

- `root`：数据集的根目录路径，应该包含子文件夹，每个子文件夹都包含一个类别的图像。
- `transform`：一个可调用的函数，接收一个PIL图片并返回一个转换后的版本。如果未提供，则不进行任何转换。
- `target_transform`：一个可调用的函数，接收一个目标（类别）并返回一个转换后的版本。如果未提供，则不进行任何转换。
- `loader`：一个可调用的函数，用于读取图像文件并返回图像的数据。默认为`default_loader`函数，可以加载JPEG、PNG、BMP和GIF图像文件。
- `is_valid_file`：一个可调用的函数，接收一个文件名并返回一个布尔值，指示该文件是否应该包含在数据集中。默认为`None`，表示所有文件都包含在数据集中。

返回值：

- 返回一个`DatasetFolder`对象，该对象包含数据集中所有图像的路径和标签。

例子：

import torchvision.datasets as datasets
import torchvision.transforms as transforms

# 加载数据集并进行数据增强（随机裁剪和水平翻转）
transform = transforms.Compose([
    transforms.RandomCrop(224),
    transforms.RandomHorizontalFlip(),
    transforms.ToTensor()
])
train_dataset = datasets.ImageFolder(root='path/to/train', transform=transform)

这个例子中，我们加载了一个名为"train"的数据集，其中包含多个子文件夹，每个子文件夹都包含一个类别的图像。我们为训练集设置了数据增强，包括随机裁剪、水平翻转和转换为张量。最后返回一个`DatasetFolder`对象，包括训练集中所有图像的路径和标签。