IniziaInizia gratis

Costruire reti convoluzionali

Fai parte di un team che sta sviluppando un sistema di previsione meteo. Come parte del sistema, verranno installate telecamere in vari luoghi per scattare foto del cielo. Il tuo compito è costruire un modello che classifichi i diversi tipi di nuvole in queste immagini, per aiutare a individuare i fronti meteorologici in arrivo.

Decidi di creare un classificatore di immagini convoluzionale. Il modello sarà composto da due parti:

  • Un estrattore di caratteristiche che apprende un vettore di feature dall'immagine di input,
  • Un classificatore che predice la classe dell'immagine in base alle feature apprese.

Sia torch sia torch.nn as nn sono già stati importati per te, quindi iniziamo!

Questo esercizio fa parte del corso

Deep Learning intermedio con PyTorch

Visualizza il corso

Esercizio pratico interattivo

Prova a risolvere questo esercizio completando il codice di esempio.

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),
            ____,
            ____,
            ____,
            ____,
        )
Modifica ed esegui il codice