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