ComeçarComece de graça

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

Ver curso

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
)
Editar e executar o código