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
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
)