带数据增强的数据集

您已经基于云朵图片构建了图像数据集，并搭建了用于分类不同云层类型的卷积模型。在训练之前，请通过添加数据增强来改造数据集，以提升模型在云层分类上的表现。

用于设置 Dataset 和 DataLoader 的代码已经为您准备好，应该看起来很熟悉。您的任务是定义要在加载输入图像时应用的变换组合。

请注意，此前为了更好地展示，您将图像调整为 128×128；现在为了加快训练，将使用更小的尺寸。稍后您会看到，64×64 足以让模型学习。