MulaiMulai sekarang secara gratis

Augmentasi data di PyTorch

Mari sertakan augmentasi data dalam Dataset Anda dan inspeksi beberapa gambar secara visual untuk memastikan transformasi yang diinginkan diterapkan.

Pertama, Anda akan menambahkan transformasi augmentasi ke train_transforms. Gunakan pembalikan horizontal acak dan rotasi dengan sudut acak antara 0 hingga 45 derajat. Kode berikut untuk membuat Dataset dan DataLoader persis sama seperti sebelumnya. Terakhir, Anda akan mengubah bentuk gambar dan menampilkannya untuk melihat apakah transformasi augmentasi baru terlihat.

Semua impor yang Anda perlukan sudah dipanggil untuk Anda:

from torchvision.datasets import ImageFolder
from torch.utils.data import DataLoader
from torchvision import transforms
import matplotlib.pyplot as plt

Saatnya mengaugmentasi beberapa foto awan!

Latihan ini adalah bagian dari kursus

Deep Learning Lanjutan dengan PyTorch

Lihat Kursus

Petunjuk latihan

  • Tambahkan dua transformasi lagi ke train_transforms untuk melakukan pembalikan horizontal acak dan kemudian rotasi dengan sudut acak antara 0 hingga 45 derajat.
  • Ubah bentuk tensor image dari DataLoader agar cocok untuk ditampilkan.
  • Tampilkan gambar.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

train_transforms = transforms.Compose([
    # Add horizontal flip and rotation
    ____,
    ____,
    transforms.ToTensor(),
    transforms.Resize((128, 128)),
])

dataset_train = ImageFolder(
  "clouds_train",
  transform=train_transforms,
)

dataloader_train = DataLoader(
  dataset_train, shuffle=True, batch_size=1
)

image, label = next(iter(dataloader_train))
# Reshape the image tensor
image = image.____.____(____, ____, ____) 
# Display the image
____
plt.show()
Edit dan Jalankan Kode