LoslegenKostenlos loslegen

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

Kurs anzeigen

Anleitung zur Übung

  • Erstelle zwei Transformationen: Die erste, um das Bild in einen Tensor umzuwandeln, und eine weitere, um das Bild auf 128 mal 128 zu skalieren. Weise sie train_transforms zu.
  • Verwende ImageFolder, um dataset_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=____,
)
Code bearbeiten und ausführen