Construcción de redes convolucionales
Formas parte de un equipo que construye un sistema de previsión meteorológica. Como parte del sistema, se instalarán cámaras en varios lugares para tomar imágenes del cielo. Tu tarea consiste en construir un modelo para clasificar los distintos tipos de nubes en estas imágenes, lo que ayudará a detectar los frentes meteorológicos que se aproximan.
Decides construir un clasificador convolucional de imágenes. El modelo constará de dos partes:
- Un extractor de características que aprende un vector de características a partir de la imagen de entrada.
- Un clasificador que predice la clase de la imagen según las características aprendidas.
Tanto torch
como torch.nn as nn
ya se han importado para ti, así que ¡manos a la obra!
Este ejercicio forma parte del curso
Aprendizaje profundo intermedio con PyTorch
Ejercicio interactivo práctico
Prueba este ejercicio completando el código de muestra.
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),
____,
____,
____,
____,
)