ComenzarEmpieza gratis

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

Ver curso

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),
            ____,
            ____,
            ____,
            ____,
        )
Editar y ejecutar código