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
Petunjuk latihan
- Tambahkan dua transformasi lagi ke
train_transformsuntuk melakukan pembalikan horizontal acak dan kemudian rotasi dengan sudut acak antara 0 hingga 45 derajat. - Ubah bentuk tensor
imagedari 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()