Bilddatensatz
Lass uns damit beginnen, ein Torch-Dataset für die Bilder zu erstellen. Du wirst es verwenden, um die Daten zu erkunden und später in ein Modell einzuspeisen.
Die Trainingsdaten für die Wolkenklassifizierung sind in der folgenden Verzeichnisstruktur gespeichert:
clouds_train
- cirriform clouds
- 539cd1c356e9c14749988a12fdf6c515.jpg
- ...
- clear sky
- cumulonimbus clouds
- cumulus clouds
- high cumuliform clouds
- stratiform clouds
- stratocumulus clouds
In clouds_train
befinden sich sieben Ordner, die jeweils eine Wolkenart (oder einen klaren Himmel) darstellen. In jedem dieser Ordner befinden sich die entsprechenden Bilddateien.
Die folgenden Imports wurden bereits für dich durchgeführt:
from torchvision.datasets import ImageFolder
from torchvision import transforms
Diese Übung ist Teil des Kurses
Deep Learning mit PyTorch für Fortgeschrittene
Anleitung zur Übung
- Erstelle zwei Transformationen: Die erste, um das Bild in einen Tensor umzuwandeln, und eine weitere, um das Bild auf
128
mal128
zu skalieren. Weise sietrain_transforms
zu. - Verwende
ImageFolder
, umdataset_train
zu definieren, und übergib dabei den Verzeichnispfad zu den Daten ("clouds_train"
) und die zuvor definierten Transformationen.
Interaktive Übung
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
# Compose transformations
train_transforms = ____([
transforms.____,
transforms.____,
])
# Create Dataset using ImageFolder
dataset_train = ____(
____,
transform=____,
)