MulaiMulai sekarang secara gratis

Himpunan data dengan augmentasi

Anda telah membangun himpunan data citra dari gambar awan dan model konvolusional untuk mengklasifikasikan berbagai tipe awan. Sebelum melatihnya, mari sesuaikan himpunan data dengan menambahkan augmentasi yang dapat meningkatkan kinerja klasifikasi awan oleh model.

Kode untuk menyiapkan Dataset dan DataLoader sudah disediakan dan seharusnya terasa familier. Tugas Anda adalah mendefinisikan komposisi transformasi yang akan diterapkan pada citra masukan saat dimuat.

Perhatikan bahwa sebelumnya Anda mengubah ukuran citra menjadi 128 × 128 agar tampil rapi, tetapi sekarang Anda akan menggunakan ukuran yang lebih kecil untuk mempercepat pelatihan. Seperti yang akan Anda lihat nanti, 64 × 64 sudah cukup besar bagi model untuk belajar.

from torchvision import transforms telah dijalankan untuk Anda, jadi mari kita mulai!

Latihan ini adalah bagian dari kursus

Deep Learning Lanjutan dengan PyTorch

Lihat Kursus

Petunjuk latihan

  • Definisikan train_transforms dengan menggabungkan lima transformasi: pembalikan horizontal acak, rotasi acak (dengan sudut dari 0 hingga 45 derajat), penyesuaian kontras otomatis acak, parsing ke tensor, dan pengubahan ukuran menjadi 64 × 64 piksel.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

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

dataset_train = ImageFolder(
  "clouds_train",
  transform=train_transforms,
)
dataloader_train = DataLoader(
  dataset_train, shuffle=True, batch_size=16
)
Edit dan Jalankan Kode