开始使用免费开始使用

带数据增强的数据集

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

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

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

from torchvision import transforms 已经为您执行完毕,开始吧!

本练习是课程的一部分

PyTorch 深度学习进阶

查看课程

练习说明

  • 通过组合五种变换来定义 train_transforms:随机水平翻转、随机旋转(角度范围 0 到 45 度)、随机自动对比度调整、解析为张量,以及调整大小为 64×64 像素。

交互式实操练习

通过完成这段示例代码来试试这个练习。

# Define transforms
train_transforms = transforms.Compose([
    ____,
    ____,
    ____,
    ____,
    ____,
])

dataset_train = ImageFolder(
  "clouds_train",
  transform=train_transforms,
)
dataloader_train = DataLoader(
  dataset_train, shuffle=True, batch_size=16
)
编辑并运行代码