Conjunto de dados com aumentos
Você já criou o conjunto de dados de imagens de nuvens e o modelo convolucional para classificar diferentes tipos de nuvens. Antes de você treiná-lo, vamos adaptar o conjunto de dados adicionando as ampliações que podem melhorar o desempenho de classificação de nuvem do modelo.
O código para configurar o conjunto de dados e o DataLoader já está preparado para você e deve parecer familiar. Sua tarefa é definir a composição das transformações que serão aplicadas às imagens de entrada à medida que forem carregadas.
Observe que antes você redimensionava as imagens para 128 por 128 para exibi-las bem, mas agora você usará imagens menores para acelerar o treinamento. Como você verá mais adiante, 64 por 64 será grande o suficiente para o modelo aprender.
from torchvision import transforms
já foi executado para você, então vamos ao que interessa!
Este exercício faz parte do curso
Aprendizagem profunda intermediária com PyTorch
Instruções do exercício
- Defina
train_transforms
compondo cinco transformações: uma inversão horizontal aleatória, rotação aleatória (por ângulo de 0 a 45 graus), ajuste de contraste automático aleatório, análise para tensor e redimensionamento para 64 por 64 pixels.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# Define transforms
train_transforms = transforms.Compose([
____,
____,
____,
____,
____,
])
dataset_train = ImageFolder(
"clouds_train",
transform=train_transforms,
)
dataloader_train = DataLoader(
dataset_train, shuffle=True, batch_size=16
)