開始使用免費開始

含資料增強的 Dataset

你已經用雲朵圖片建立了影像資料集,並實作了用來分類不同雲朵類型的卷積模型。開始訓練之前,先為資料集加入可能提升雲朵分類效能的資料增強。

設定 Dataset 與 DataLoader 的程式碼已經為你準備好,而且應該很眼熟。你的工作是定義一組要套用在載入中輸入影像上的轉換組合。

先前你將影像縮放為 128×128 以便好看地顯示;但現在為了加快訓練,會改用更小的影像。接下來你會看到,64×64 對模型學習已經足夠。

from torchvision import transforms 已經替你執行好了,開始動手吧!

本練習屬於課程

Intermediate Deep Learning with 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
)
編輯並執行程式碼