LoslegenKostenlos loslegen

Datensatz mit Augmentierungen

Du hast bereits den Bilddatensatz aus Wolkenbildern und das Convolutional-Modell zur Klassifizierung verschiedener Wolkentypen erstellt. Bevor du es trainierst, lass uns den Datensatz durch Hinzufügen von Augmentierungen anpassen, die die Wolkenklassifizierungsleistung des Modells verbessern könnten.

Der Code zum Einrichten des Datasets und DataLoaders ist bereits für dich vorbereitet und sollte dir bekannt vorkommen. Deine Aufgabe ist es, die Zusammensetzung der Transformationen zu definieren, die auf die Eingangsbilder beim Laden angewendet werden.

Beachte, dass du zuvor die Bilder auf 128 mal 128 skaliert hast, um sie schön anzuzeigen, aber jetzt wirst du kleinere verwenden, um das Training zu beschleunigen. Wie du später sehen wirst, wird 64 mal 64 groß genug sein, damit das Modell lernen kann.

from torchvision import transforms wurde bereits für dich ausgeführt, lass uns also loslegen!

Diese Übung ist Teil des Kurses

Deep Learning mit PyTorch für Fortgeschrittene

Kurs anzeigen

Anleitung zur Übung

  • Definiere train_transforms durch die Kombination von fünf Transformationen: eine zufällige horizontale Spiegelung, zufällige Rotation (mit einem Winkel von 0 bis 45 Grad), zufällige automatische Kontrastanpassung, Umwandlung in einen Tensor und Größenänderung auf 64 mal 64 Pixel.

Interaktive Übung

Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.

# Define transforms
train_transforms = transforms.Compose([
    ____,
    ____,
    ____,
    ____,
    ____,
])

dataset_train = ImageFolder(
  "clouds_train",
  transform=train_transforms,
)
dataloader_train = DataLoader(
  dataset_train, shuffle=True, batch_size=16
)
Code bearbeiten und ausführen