CommencerCommencer gratuitement

Créer des réseaux convolutionnels

Vous faites partie d’une équipe qui construit un système de prévision météorologique. Dans le cadre du système, des caméras seront installées à différents endroits pour photographier le ciel. Votre tâche est de créer un modèle capable de classer les différents types de nuages sur ces images, afin d’aider à repérer les fronts météorologiques qui approchent.

Vous décidez de construire un classificateur d’images convolutionnel. Le modèle comportera deux parties :

  • Un extracteur de caractéristiques qui apprend un vecteur de caractéristiques à partir de l’image d’entrée,
  • Un classificateur qui prédit la classe de l’image à partir des caractéristiques apprises.

torch et torch.nn as nn ont déjà été importés pour vous, alors allons-y !

Cet exercice fait partie du cours

Deep learning intermédiaire avec PyTorch

Afficher le cours

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

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),
            ____,
            ____,
            ____,
            ____,
        )
Modifier et exécuter le code