26、石头剪刀布图像分类：数据准备、卷积与模型优化

石头剪刀布图像分类：数据准备、卷积与模型优化

1. 数据准备

本次数据准备工作要求更高，需要对图像进行标准化处理，同时可以使用 ImageFolder 数据集。

1.1 ImageFolder 数据集

ImageFolder 并非实际的数据集，而是一个通用数据集，只要图像被正确组织到子文件夹中，每个子文件夹以类别命名并包含相应图像，就可以使用。以石头剪刀布数据集为例，训练集的 rps 文件夹内有三个子文件夹，分别为 rock 、 paper 和 scissors ，每个子文件夹包含 840 张对应类别的图像。

ImageFolder 数据集只需要根文件夹（这里是 rps 文件夹），还可以接受四个可选参数：
- transform ：指定对每张图像应用的变换，如数据增强变换。
- target_transform ：若目标也是图像（如分割任务），该参数才有意义。
- loader ：当使用 PIL 无法处理的特殊图像格式时，用于从给定路径加载图像的函数。
- is_valid_file ：检查文件是否损坏的函数。

创建临时数据集的代码如下：

from torchvi