ComeçarComece gratuitamente

Criação de redes convolucionais

Você faz parte de uma equipe que está construindo um sistema de previsão do tempo. Como parte do sistema, câmeras serão instaladas em vários locais para tirar fotos do céu. Sua tarefa é criar um modelo para classificar os diferentes tipos de nuvens nessas imagens, o que ajudará a identificar as frentes meteorológicas que se aproximam.

Você decide criar um classificador de imagem convolucional. O modelo consistirá em duas partes:

  • Um extrator de recursos que aprende um vetor de recursos a partir da imagem de entrada,
  • Um classificador que prevê a classe da imagem com base nos recursos aprendidos.

Tanto o torch quanto o torch.nn as nn já foram importados para você, então vamos ao que interessa!

Este exercício faz parte do curso

Aprendizagem profunda intermediária com PyTorch

Ver Curso

Exercício interativo prático

Experimente este exercício preenchendo este código de exemplo.

class Net(nn.Module):
    def __init__(self, num_classes):
        super().__init__()
        # Define feature extractor
        self.feature_extractor = nn.Sequential(
            nn.Conv2d(3, 32, kernel_size=3, padding=1),
            nn.ELU(),
            nn.MaxPool2d(kernel_size=2),
            ____,
            ____,
            ____,
            ____,
        )
Editar e executar código