Aan de slagGa gratis aan de slag

Convolutionele netwerken bouwen

Je zit in een team dat een weersvoorspellingssysteem bouwt. Als onderdeel van het systeem worden op verschillende locaties camera's geplaatst om foto's van de lucht te maken. Jouw taak is een model te bouwen dat verschillende wolkentypes in deze foto's classificeert, zodat naderende fronten sneller worden gesignaleerd.

Je besluit een convolutionele beeldclassifier te bouwen. Het model bestaat uit twee delen:

  • Een feature extractor die een vector met kenmerken uit de invoerafbeelding leert,
  • Een classifier die de klasse van de afbeelding voorspelt op basis van de geleerde kenmerken.

Zowel torch als torch.nn as nn zijn al voor je geïmporteerd, dus laten we aan de slag gaan!

Deze oefening maakt deel uit van de cursus

Gevorderde Deep Learning met PyTorch

Cursus bekijken

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

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),
            ____,
            ____,
            ____,
            ____,
        )
Code bewerken en uitvoeren