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 merupakan bagian dari kursus
Deep Learning Lanjutan dengan PyTorch
Instruksi 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 langsung praktik
Cobalah latihan ini dengan melengkapi kode contoh ini.
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()