Himpunan data gambar
Mari mulai dengan membangun Torch Dataset untuk gambar. Anda akan menggunakannya untuk menelusuri data dan, nanti, untuk memasukkannya ke dalam model.
Data pelatihan untuk tugas klasifikasi awan disimpan dalam struktur direktori berikut:
clouds_train
- cirriform clouds
- 539cd1c356e9c14749988a12fdf6c515.jpg
- ...
- clear sky
- cumulonimbus clouds
- cumulus clouds
- high cumuliform clouds
- stratiform clouds
- stratocumulus clouds
Ada tujuh folder di dalam clouds_train, masing-masing merepresentasikan satu jenis awan (atau langit cerah). Di dalam setiap folder tersebut terdapat berkas gambar yang sesuai.
Impor berikut sudah disiapkan untuk Anda:
from torchvision.datasets import ImageFolder
from torchvision import transforms
Latihan ini adalah bagian dari kursus
Deep Learning Lanjutan dengan PyTorch
Petunjuk latihan
- Rangkai dua transformasi: pertama, mengubah gambar menjadi tensor, dan satu lagi untuk mengubah ukuran gambar menjadi
128kali128, lalu tetapkan ketrain_transforms. - Gunakan
ImageFolderuntuk mendefinisikandataset_train, dengan memberikan path direktori ke data ("clouds_train") dan transformasi yang telah didefinisikan sebelumnya.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Compose transformations
train_transforms = ____([
transforms.____,
transforms.____,
])
# Create Dataset using ImageFolder
dataset_train = ____(
____,
transform=____,
)